提交 1e0efdb6 authored 作者: Olivier Delalleau's avatar Olivier Delalleau

Fixed circular import issue

Sorry, I should have tested more thoroughly this stuff with GPU before pushing it :/
上级 7ccfbb01
...@@ -154,14 +154,6 @@ if cuda_available: ...@@ -154,14 +154,6 @@ if cuda_available:
cuda_available = False cuda_available = False
cuda_initialization_error_message = e.message 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): class GpuOp(theano.gof.Op):
...@@ -186,6 +178,15 @@ class GpuOp(theano.gof.Op): ...@@ -186,6 +178,15 @@ class GpuOp(theano.gof.Op):
compute_map, no_recycling) 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: if cuda_available:
# check if their is an old cuda_ndarray that was loading instead of the one # check if their is an old cuda_ndarray that was loading instead of the one
# we compiled! # we compiled!
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论