提交 3b70da60 authored 作者: Frederic Bastien's avatar Frederic Bastien

when we test on the gpu, we skip test that don't use float32.

上级 9b8d4456
## PENDING REWRITE OF tensor_opt.py ## PENDING REWRITE OF tensor_opt.py
import unittest
from theano import gof from theano import gof
from theano.tensor.opt import * from theano.tensor.opt import *
...@@ -934,6 +933,9 @@ class test_fusion(unittest.TestCase): ...@@ -934,6 +933,9 @@ class test_fusion(unittest.TestCase):
fail3=[] fail3=[]
fail4=[] fail4=[]
for id, [g, sym_inputs, val_inputs, nb_elemwise, answer, out_dtype] in enumerate(cases): for id, [g, sym_inputs, val_inputs, nb_elemwise, answer, out_dtype] in enumerate(cases):
if gpu and out_dtype!='float32':
print "Skip test %d as the gpu code currently support only float32" % id
continue
print "new cases", id print "new cases", id
if shared_fn == None: if shared_fn == None:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论