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

fix test crash

上级 5386d3f9
...@@ -179,6 +179,7 @@ class test_GpuCAReduceCuda(test_GpuCAReduceCPY): ...@@ -179,6 +179,7 @@ class test_GpuCAReduceCuda(test_GpuCAReduceCPY):
return return
def setUp(self): def setUp(self):
super(test_GpuCAReduceCuda, self).setUp()
dev = theano.sandbox.gpuarray.init_dev.device dev = theano.sandbox.gpuarray.init_dev.device
if not dev.startswith('cuda'): if not dev.startswith('cuda'):
raise SkipTest("Cuda specific tests") raise SkipTest("Cuda specific tests")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论