提交 d0e0274f authored 作者: Frederic's avatar Frederic

compile cuda_ndarray.cu with -O3 flags.

上级 7cc1d895
...@@ -164,7 +164,7 @@ if compile_cuda_ndarray and cuda_available: ...@@ -164,7 +164,7 @@ if compile_cuda_ndarray and cuda_available:
code, code,
location=cuda_ndarray_loc, location=cuda_ndarray_loc,
include_dirs=[cuda_path], libs=['cublas'], include_dirs=[cuda_path], libs=['cublas'],
preargs=compiler.compile_args()) preargs=['-O3'] + compiler.compile_args())
from cuda_ndarray.cuda_ndarray import * from cuda_ndarray.cuda_ndarray import *
except Exception, e: except Exception, 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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论