remove dependence of scikits.cuda from rbf kernel

This commit is contained in:
Zhenwen Dai 2014-06-26 14:12:19 +01:00
parent 92fecbb4ef
commit 1512590a8c
2 changed files with 8 additions and 4 deletions

View file

@ -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

View file

@ -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)