From 7a35c9a6ade5a6d8d64d1b0665072b86af714cff Mon Sep 17 00:00:00 2001
From: Zhenwen Dai <z.dai@sheffield.ac.uk>
Date: Mon, 23 Jun 2014 12:02:36 +0100
Subject: [PATCH] fix pickle for RBF GPU kernel

---
 GPy/kern/_src/rbf.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/GPy/kern/_src/rbf.py b/GPy/kern/_src/rbf.py
index a6f31ff2..4aa3ed74 100644
--- a/GPy/kern/_src/rbf.py
+++ b/GPy/kern/_src/rbf.py
@@ -38,6 +38,15 @@ class RBF(Stationary):
     def dK_dr(self, r):
         return -r*self.K_of_r(r)
 
+    def __getstate__(self):
+        dc = super(RBF, self).__getstate__()
+        if self.useGPU:
+            dc['psicomp'] = PSICOMP_RBF()
+        return dc
+ 
+    def __setstate__(self, state):
+        return super(RBF, self).__setstate__(state)
+
     #---------------------------------------#
     #             PSI statistics            #
     #---------------------------------------#