提交 389c4aba authored 作者: Frederic's avatar Frederic

Tmp fix to make cumem work with the reduced visibility

上级 1e0992f7
...@@ -184,7 +184,9 @@ if compile_cuda_ndarray and cuda_available: ...@@ -184,7 +184,9 @@ if compile_cuda_ndarray and cuda_available:
location=cuda_ndarray_loc, location=cuda_ndarray_loc,
include_dirs=[cuda_path], include_dirs=[cuda_path],
libs=[config.cublas.lib], libs=[config.cublas.lib],
preargs=['-O3'] + compiler.compile_args()) preargs=['-O3'] + compiler.compile_args(),
hide_symbols=False,
)
from cuda_ndarray.cuda_ndarray import * from cuda_ndarray.cuda_ndarray import *
except Exception as e: except Exception as e:
_logger.error("Failed to compile cuda_ndarray.cu: %s", str(e)) _logger.error("Failed to compile cuda_ndarray.cu: %s", str(e))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论