提交 ab05b05d authored 作者: Frederic's avatar Frederic

correctly skip cuda test when the cuda back-end is not available.

上级 eec8002e
...@@ -11,11 +11,11 @@ import theano ...@@ -11,11 +11,11 @@ import theano
from theano.tests import unittest_tools as utt from theano.tests import unittest_tools as utt
import theano.sandbox.cuda as cuda import theano.sandbox.cuda as cuda
from theano.sandbox.cuda import basic_ops
if cuda.cuda_available == False: if cuda.cuda_available == False:
raise SkipTest('Optional package cuda disabled') raise SkipTest('Optional package cuda disabled')
from theano.sandbox.cuda import basic_ops
from theano.sandbox.cuda.type import CudaNdarrayType from theano.sandbox.cuda.type import CudaNdarrayType
if theano.config.mode=='FAST_COMPILE': if theano.config.mode=='FAST_COMPILE':
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论