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

Initialize CULA at the last moment

上级 a8371735
...@@ -15,12 +15,6 @@ except (ImportError, OSError): ...@@ -15,12 +15,6 @@ except (ImportError, OSError):
pass pass
cula_initialized = False 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): class GpuSolve(GpuOp):
...@@ -62,6 +56,12 @@ class GpuSolve(GpuOp): ...@@ -62,6 +56,12 @@ class GpuSolve(GpuOp):
no_recycling=[]): no_recycling=[]):
from theano.misc.pycuda_utils import to_gpuarray 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] inputs = [storage_map[v] for v in node.inputs]
outputs = [storage_map[v] for v in node.outputs] outputs = [storage_map[v] for v in node.outputs]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论