apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-30 14:35:15 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

resolve the problem of pickling a gpu kernel

Browse source
This commit is contained in:
Zhenwen Dai 2014-10-17 15:29:24 +01:00
parent e9d33ddc7e
commit deb48407ff
1 changed files with 3 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

10
GPy/kern/_src/rbf.py
View file

@ -34,16 +34,12 @@ class RBF(Stationary):
def __getstate__(self): def __getstate__(self):
dc = super(RBF, self).__getstate__() dc = super(RBF, self).__getstate__()
# if self.useGPU: if self.useGPU:
# dc['psicomp'] = PSICOMP_RBF() dc['psicomp'] = PSICOMP_RBF()
# dc['useGPU'] = False dc['useGPU'] = False
return dc return dc
def __setstate__(self, state): 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) return super(RBF, self).__setstate__(state)
def spectrum(self, omega): def spectrum(self, omega):