提交 70c1f770 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix fusion test on the gpu.

上级 8ca64276
......@@ -786,7 +786,7 @@ class test_fusion(unittest.TestCase):
fail3=[]
fail4=[]
for id, [g, sym_inputs, val_inputs, nb_elemwise, answer, out_dtype] in enumerate(cases):
if gpu and out_dtype!='float32':
if gpu and (out_dtype!='float32' or any(i.dtype != 'float32' for i in g.owner.inputs)):
print "Skip test %d as the gpu code currently support only float32" % id
continue
print "new cases", id
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论