From deb48407ff2304c2e6feb32c65331096ec6afc6e Mon Sep 17 00:00:00 2001 From: Zhenwen Dai Date: Fri, 17 Oct 2014 15:29:24 +0100 Subject: [PATCH] resolve the problem of pickling a gpu kernel --- GPy/kern/_src/rbf.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/GPy/kern/_src/rbf.py b/GPy/kern/_src/rbf.py index 677491d5..1723494f 100644 --- a/GPy/kern/_src/rbf.py +++ b/GPy/kern/_src/rbf.py @@ -34,16 +34,12 @@ class RBF(Stationary): def __getstate__(self): dc = super(RBF, self).__getstate__() -# if self.useGPU: -# dc['psicomp'] = PSICOMP_RBF() -# dc['useGPU'] = False + if self.useGPU: + dc['psicomp'] = PSICOMP_RBF() + dc['useGPU'] = False return dc def __setstate__(self, state): - from ...util.gpu_init import gpu_initialized - if state['useGPU'] and not gpu_initialized: - state['useGPU'] = False - state['psicomp'] = PSICOMP_RBF() return super(RBF, self).__setstate__(state) def spectrum(self, omega):