apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-07 19:12:40 +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:22:38 +00:00
parent 0dd52981d0
commit a77a675549
3 changed files with 29 additions and 28 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

@ -5,7 +5,6 @@ The module for psi-statistics for RBF kernel
import numpy as np
from paramz.caching import Cache_this
from . import PSICOMP_RBF
from ....util import gpu_init
gpu_code = """
// define THREADNUM
@ -238,8 +237,6 @@ class PSICOMP_RBF_GPU(PSICOMP_RBF):
self.fall_back = PSICOMP_RBF()
from pycuda.compiler import SourceModule
from ....util.gpu_init import initGPU
initGPU()
self.GPU_direct = GPU_direct
self.gpuCache = None

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

@ -287,8 +287,6 @@ class PSICOMP_SSRBF_GPU(PSICOMP_RBF):
def __init__(self, threadnum=128, blocknum=15, GPU_direct=False):
from pycuda.compiler import SourceModule
from ....util.gpu_init import initGPU
initGPU()
self.GPU_direct = GPU_direct
self.gpuCache = None