提交 eb4d03d8 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix import when cuda is not available.

上级 26cd1b81
...@@ -14,8 +14,9 @@ from theano.tensor import zeros_like, sqrt, log, sin, cos, join ...@@ -14,8 +14,9 @@ from theano.tensor import zeros_like, sqrt, log, sin, cos, join
from theano.compile import optdb from theano.compile import optdb
from theano.gof import local_optimizer from theano.gof import local_optimizer
from theano.sandbox.cuda.opt import register_opt as gpu_register_opt from theano.sandbox.cuda import cuda_enabled
from theano.sandbox.cuda import cuda_enabled, CudaNdarrayType #, gpu_from_host, host_from_gpu, CudaNdarrayType if cuda_enabled:
from theano.sandbox.cuda import CudaNdarrayType
def mulmod(a, b, c, m): def mulmod(a, b, c, m):
r = numpy.int32((numpy.int64(a)*b + c) % m) r = numpy.int32((numpy.int64(a)*b + c) % m)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论