提交 5794f14a authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix compilation crash when cuda is not in CPATH.

上级 fc12422d
......@@ -58,7 +58,8 @@ class GpuConnectionistTemporalClassification(gof.COp):
return ["warpctc", "gpuarray"]
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.ctc.root != '':
dirs.append(os.path.join(config.ctc.root, "include"))
return dirs
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论