mirror of
https://github.com/SheffieldML/GPy.git
synced 2026-06-14 15:25:15 +02:00
remove dependence of scikits.cuda from rbf kernel
This commit is contained in:
parent
92fecbb4ef
commit
1512590a8c
2 changed files with 8 additions and 4 deletions
|
|
@ -17,7 +17,7 @@ except:
|
|||
pass
|
||||
|
||||
try:
|
||||
if MPI_enabled and MPI.COMM_WORLD.size>1:
|
||||
if MPI_enabled: #and MPI.COMM_WORLD.size>1:
|
||||
from .parallel import get_id_within_node
|
||||
gpuid = get_id_within_node()
|
||||
import pycuda.driver
|
||||
|
|
|
|||
|
|
@ -12,9 +12,6 @@ from ..util import gpu_init
|
|||
try:
|
||||
from pycuda.reduction import ReductionKernel
|
||||
from pycuda.elementwise import ElementwiseKernel
|
||||
import scikits.cuda.linalg as culinalg
|
||||
from scikits.cuda import cublas
|
||||
from scikits.cuda.cula import culaExceptions
|
||||
|
||||
# log|A| for A is a low triangle matrix
|
||||
# logDiagSum(A, A.shape[0]+1)
|
||||
|
|
@ -64,6 +61,13 @@ try:
|
|||
except:
|
||||
pass
|
||||
|
||||
try:
|
||||
import scikits.cuda.linalg as culinalg
|
||||
from scikits.cuda import cublas
|
||||
from scikits.cuda.cula import culaExceptions
|
||||
except:
|
||||
pass
|
||||
|
||||
def jitchol(A, L, cublas_handle, maxtries=5):
|
||||
try:
|
||||
cublas.cublasDcopy(cublas_handle, A.size, A.gpudata, 1, L.gpudata, 1)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue