apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-08 03:22:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

enable rbf gpu to support psi2n

Browse source
This commit is contained in:
Zhenwen Dai 2015-11-15 23:11:00 +00:00
parent 3a3f6cee44
commit 4f1328980c
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
GPy/kern/src/psi_comp/rbf_psi_gpucomp.py
View file

@ -360,7 +360,10 @@ class PSICOMP_RBF_GPU(PSICOMP_RBF):
if self.GPU_direct:
return psi0, psi1_gpu, psi2_gpu
else:
return psi0, psi1_gpu.get(), psi2_gpu.get()
if return_psi2_n:
return psi0, psi1_gpu.get(), psi2n_gpu.get()
else:
return psi0, psi1_gpu.get(), psi2_gpu.get()
def psiDerivativecomputations(self, kern, dL_dpsi0, dL_dpsi1, dL_dpsi2, Z, variational_posterior):
try: