提交 09115c71 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix a test following theano internal interface change.

上级 72f97e3a
......@@ -892,7 +892,8 @@ class test_fusion(unittest.TestCase):
import theano.sandbox.cuda as cuda
if not cuda.cuda_available:
raise SkipTest("cuda not available")
if cuda.opt.int_size == 4:
sizes = cuda.opt.get_device_type_sizes()
if sizes['int_size'] == 4:
shp=(5,5,5,5)
else:
shp=(5,5,5)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论