提交 e32e9d5f authored 作者: Frederic Bastien's avatar Frederic Bastien

fix a test when floatX=float32.

上级 d44742fd
...@@ -4160,7 +4160,7 @@ class test_broadcast(unittest.TestCase): ...@@ -4160,7 +4160,7 @@ class test_broadcast(unittest.TestCase):
x = matrix() x = matrix()
y = addbroadcast(x,0) y = addbroadcast(x,0)
f = theano.function([x], y.shape) f = theano.function([x], y.shape)
f(numpy.zeros((1,5))) f(numpy.zeros((1,5), dtype=config.floatX))
topo = f.maker.env.toposort() topo = f.maker.env.toposort()
if theano.config.mode != 'FAST_COMPILE': if theano.config.mode != 'FAST_COMPILE':
assert len(topo) == 3 assert len(topo) == 3
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论