提交 bfc8fc92 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix import when not into the theano/sandbox/cuda directory.

上级 e6d2f0c7
......@@ -47,7 +47,7 @@ def nvcc_module_compile_str(module_name, src_code, location=None, include_dirs=[
preargs.append('-fPIC')
no_opt = False
cuda_root = config.CUDA_ROOT
include_dirs = std_include_dirs() + include_dirs
include_dirs = std_include_dirs() + include_dirs + [os.path.split(__file__)[0]]
libs = std_libs() + ['cudart'] + libs
lib_dirs = std_lib_dirs() + lib_dirs
if cuda_root:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论