提交 3b8cde2b authored 作者: Pascal Lamblin's avatar Pascal Lamblin 提交者: Caglar

Initialize CULA at the last moment

上级 a8371735
......@@ -15,12 +15,6 @@ except (ImportError, OSError):
pass
cula_initialized = False
if cula_available and cula and not cula_initialized:
try:
cula.culaInitialize()
cula_initialized = True
except:
warnings.warn("Initialization of cula failed.")
class GpuSolve(GpuOp):
......@@ -62,6 +56,12 @@ class GpuSolve(GpuOp):
no_recycling=[]):
from theano.misc.pycuda_utils import to_gpuarray
# Initialize CULA the first time it is needed
global cula_initialized
if cula_available and cula and not cula_initialized:
cula.culaInitialize()
cula_initialized = True
inputs = [storage_map[v] for v in node.inputs]
outputs = [storage_map[v] for v in node.outputs]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论