apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-04-29 23:06:22 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix gpu initialziation

Browse source
This commit is contained in:
Zhenwen Dai 2016-02-09 17:25:25 +00:00
parent a77a675549
commit 3917d9cdba
2 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -237,6 +237,7 @@ class PSICOMP_RBF_GPU(PSICOMP_RBF):
self.fall_back = PSICOMP_RBF()
from pycuda.compiler import SourceModule
import GPy.util.gpu_init
self.GPU_direct = GPU_direct
self.gpuCache = None

2
GPy/kern/src/psi_comp/ssrbf_psi_gpucomp.py
View file

@ -7,7 +7,6 @@ import numpy as np
from paramz.caching import Cache_this
from . import PSICOMP_RBF
gpu_code = """
// define THREADNUM
@ -287,6 +286,7 @@ class PSICOMP_SSRBF_GPU(PSICOMP_RBF):
def __init__(self, threadnum=128, blocknum=15, GPU_direct=False):
from pycuda.compiler import SourceModule
import GPy.util.gpu_init
self.GPU_direct = GPU_direct
self.gpuCache = None