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

[TEST] Fix test in 32 bit python.

上级 d22242b9
......@@ -3888,7 +3888,8 @@ class T_local_reduce(unittest.TestCase):
assert isinstance(topo[-1].op, T.Elemwise), out
# Test different axis for the join and the reduction
A = theano.shared(numpy.array([1, 2, 3, 4, 5]))
# We must force the dtype, of otherwise, this tests will fail in 32 bit system
A = theano.shared(numpy.array([1, 2, 3, 4, 5], dtype='int64'))
f = theano.function([], T.sum(T.stack(A, A), axis=0), mode=self.mode)
assert numpy.allclose(f(), [2, 4, 6, 8, 10])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论