提交 d7ae3a52 authored 作者: notoraptor's avatar notoraptor

Add cuda include path for magma ops

上级 f636d83a
......@@ -498,7 +498,7 @@ class GpuMagmaBase(COp):
'gpuarray_helper.h', 'magma.h']
def c_header_dirs(self):
dirs = [gpuarray_helper_inc_dir(), pygpu.get_include()]
dirs = [gpuarray_helper_inc_dir(), pygpu.get_include(), config.cuda.include_path]
if config.magma.include_path:
dirs.append(config.magma.include_path)
return dirs
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论