提交 95d19a6a authored 作者: lamblin's avatar lamblin

Merge pull request #463 from delallea/fix_disable_cuda

Fixed circular import issue
......@@ -154,14 +154,6 @@ if cuda_available:
cuda_available = False
cuda_initialization_error_message = e.message
# We must do those import to be able to create the full doc when
# nvcc is not available
from theano.sandbox.cuda.var import (CudaNdarrayVariable,
CudaNdarrayConstant,
CudaNdarraySharedVariable,
float32_shared_constructor)
from theano.sandbox.cuda.type import CudaNdarrayType
class GpuOp(theano.gof.Op):
......@@ -186,6 +178,15 @@ class GpuOp(theano.gof.Op):
compute_map, no_recycling)
# We must do those import to be able to create the full doc when
# nvcc is not available
from theano.sandbox.cuda.var import (CudaNdarrayVariable,
CudaNdarrayConstant,
CudaNdarraySharedVariable,
float32_shared_constructor)
from theano.sandbox.cuda.type import CudaNdarrayType
if cuda_available:
# check if their is an old cuda_ndarray that was loading instead of the one
# we compiled!
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论