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