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

Fix the cula import error.

上级 413f0c53
......@@ -49,7 +49,12 @@ class GpuSolve(GpuOp):
# Initialize CULA the first time it is needed
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_initialized = True
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论