提交 117a0aa2 authored 作者: Caglar's avatar Caglar

Fix the cula import error.

上级 413f0c53
...@@ -49,7 +49,12 @@ class GpuSolve(GpuOp): ...@@ -49,7 +49,12 @@ class GpuSolve(GpuOp):
# Initialize CULA the first time it is needed # Initialize CULA the first time it is needed
global cula_initialized global cula_initialized
if cula_available and cula and not cula_initialized:
if not cula_available:
raise RuntimeError('Cula is not available and '
'GpuSolve Op can not be called.')
if not cula_initialized:
cula.culaInitialize() cula.culaInitialize()
cula_initialized = True cula_initialized = True
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论