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

fix test crash

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