apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-10 12:32:40 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

[GPU] bug fix

Browse source
This commit is contained in:
Zhenwen Dai 2014-04-03 12:29:44 +01:00
parent bb5c41f64c
commit 8c4507d9f1
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
GPy/kern/_src/psi_comp/ssrbf_psi_gpucomp.py
View file

@ -260,11 +260,11 @@ class PSICOMP_SSRBF(object):
self.gpuCache = None self.gpuCache = None
def _initGPUCache(self, N, M, Q): def _initGPUCache(self, N, M, Q):
if self.gpuCache and self.gpuCacheAll['mu_gpu'].shape[0]!=N: if self.gpuCache and self.gpuCache['mu_gpu'].shape[0]!=N:
self._releaseMemory() self._releaseMemory()
if self.gpuCache == None: if self.gpuCache == None:
self.gpuCacheAll = { self.gpuCache = {
'l_gpu' :gpuarray.empty((Q,),np.float64,order='F'), 'l_gpu' :gpuarray.empty((Q,),np.float64,order='F'),
'Z_gpu' :gpuarray.empty((M,Q),np.float64,order='F'), 'Z_gpu' :gpuarray.empty((M,Q),np.float64,order='F'),
'mu_gpu' :gpuarray.empty((N,Q),np.float64,order='F'), 'mu_gpu' :gpuarray.empty((N,Q),np.float64,order='F'),
@ -306,11 +306,11 @@ class PSICOMP_SSRBF(object):
} }
def _releaseMemory(self): def _releaseMemory(self):
if not self.gpuCacheAll: if not self.gpuCache:
for k,v in self.gpuCacheAll: for k,v in self.gpuCache:
v.gpudata.free() v.gpudata.free()
del v del v
self.gpuCacheAll = None del self.gpuCache
self.gpuCache = None self.gpuCache = None
def psicomputations(self, variance, lengthscale, Z, mu, S, gamma): def psicomputations(self, variance, lengthscale, Z, mu, S, gamma):