提交 b9cbe844 authored 作者: Frederic's avatar Frederic

Allow to reuse the NVCC_compiler without cuda_ndarray lib.

Needed for the new gpuarray.
上级 de5e06e7
......@@ -347,8 +347,6 @@ class NVCC_compiler(object):
cmd.append(os.path.split(cppfilename)[-1])
cmd.extend(['-L%s' % ldir for ldir in lib_dirs])
cmd.extend(['-l%s' % l for l in libs])
if module_name != 'cuda_ndarray':
cmd.append("-lcuda_ndarray")
if sys.platform == 'darwin':
cmd.extend(darwin_python_lib.split())
......
......@@ -400,7 +400,7 @@ class CudaNdarrayType(Type):
def c_libraries(self):
# returning cublas because the cuda_ndarray.cuh header
# includes calls to SetVector and cublasGetError
return ['cudart', 'cublas']
return ['cudart', 'cublas', 'cuda_ndarray']
def c_support_code(cls):
return ""
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论