Commit 9cb886b8 authored by Ryan Gutenkunst's avatar Ryan Gutenkunst
Browse files

Move cusparse into dadi repository

parent 1419cf50
......@@ -78,10 +78,10 @@ def _inject_mutations_5D_valcalc(dt, xx, yy, zz, aa, bb, theta0, frozen1, frozen
import pycuda
import pycuda.gpuarray as gpuarray
from skcuda.cusparse import cusparseDgtsvInterleavedBatch_bufferSizeExt, cusparseDgtsvInterleavedBatch
import dadi.cuda
from dadi.cuda import cusparse_handle, _grid, _block, transpose_gpuarray
from dadi.cuda.cusparse import cusparseDgtsvInterleavedBatch_bufferSizeExt, cusparseDgtsvInterleavedBatch
from . import kernels
def _two_pops_const_params(phi, xx,
......
......@@ -2,7 +2,7 @@ import atexit
import pycuda.autoinit
from pycuda.tools import clear_context_caches, make_default_context
from skcuda.cusparse import cusparseCreate, cusparseDestroy
from dadi.cuda.cusparse import cusparseCreate, cusparseDestroy
from skcuda.cublas import cublasCreate, cublasDestroy, cublasDgeam
ctx = make_default_context()
......
This diff is collapsed.
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment