apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-08 11:32:39 +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

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

@ -359,6 +359,9 @@ class PSICOMP_RBF_GPU(PSICOMP_RBF):
if self.GPU_direct: if self.GPU_direct:
return psi0, psi1_gpu, psi2_gpu return psi0, psi1_gpu, psi2_gpu
else:
if return_psi2_n:
return psi0, psi1_gpu.get(), psi2n_gpu.get()
else: else:
return psi0, psi1_gpu.get(), psi2_gpu.get() return psi0, psi1_gpu.get(), psi2_gpu.get()