提交 25d68bd6 authored 作者: Frederic Bastien's avatar Frederic Bastien

skip test if cuda is not available.

上级 59a43850
...@@ -848,6 +848,9 @@ class test_fusion(unittest.TestCase): ...@@ -848,6 +848,9 @@ class test_fusion(unittest.TestCase):
else: else:
mode = theano.compile.mode.get_default_mode().including('local_elemwise_fusion','canonicalize','gpu') mode = theano.compile.mode.get_default_mode().including('local_elemwise_fusion','canonicalize','gpu')
import theano.sandbox.cuda as cuda import theano.sandbox.cuda as cuda
if not cuda.cuda_available:
raise SkipTest("cuda not available")
self.do(mode, cuda.float32_shared_constructor, shp, gpu=True) self.do(mode, cuda.float32_shared_constructor, shp, gpu=True)
def speed_fusion(self, shared_fn = shared, gpu = False, s=None): def speed_fusion(self, shared_fn = shared, gpu = False, s=None):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论