提交 42907a0c authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #4099 from JohnZed/master

Fix stale import of cuda_enabled in rng_mrg.py (#4098)
...@@ -21,8 +21,9 @@ from theano.compile import optdb ...@@ -21,8 +21,9 @@ from theano.compile import optdb
from theano.gof import local_optimizer from theano.gof import local_optimizer
from . import multinomial from . import multinomial
from theano.sandbox.cuda import cuda_available, cuda_enabled, GpuOp import theano.sandbox.cuda
if cuda_available: from theano.sandbox.cuda import GpuOp
if theano.sandbox.cuda.cuda_available:
from theano.sandbox.cuda import (CudaNdarrayType, from theano.sandbox.cuda import (CudaNdarrayType,
float32_shared_constructor) float32_shared_constructor)
...@@ -1091,7 +1092,7 @@ class MRG_RandomStreams(object): ...@@ -1091,7 +1092,7 @@ class MRG_RandomStreams(object):
self.set_rstate(seed) self.set_rstate(seed)
if use_cuda is None: if use_cuda is None:
self.use_cuda = cuda_enabled self.use_cuda = theano.sandbox.cuda.cuda_enabled
else: else:
self.use_cuda = use_cuda self.use_cuda = use_cuda
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论