From 3917d9cdbafb4c3a793a27709fb6dab94e6ed2e4 Mon Sep 17 00:00:00 2001 From: Zhenwen Dai Date: Tue, 9 Feb 2016 17:25:25 +0000 Subject: [PATCH] fix gpu initialziation --- GPy/kern/src/psi_comp/rbf_psi_gpucomp.py | 1 + GPy/kern/src/psi_comp/ssrbf_psi_gpucomp.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/GPy/kern/src/psi_comp/rbf_psi_gpucomp.py b/GPy/kern/src/psi_comp/rbf_psi_gpucomp.py index e9268a3e..4eda1b32 100644 --- a/GPy/kern/src/psi_comp/rbf_psi_gpucomp.py +++ b/GPy/kern/src/psi_comp/rbf_psi_gpucomp.py @@ -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 diff --git a/GPy/kern/src/psi_comp/ssrbf_psi_gpucomp.py b/GPy/kern/src/psi_comp/ssrbf_psi_gpucomp.py index 62876785..04063dcc 100644 --- a/GPy/kern/src/psi_comp/ssrbf_psi_gpucomp.py +++ b/GPy/kern/src/psi_comp/ssrbf_psi_gpucomp.py @@ -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