提交 0bf0e5a3 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #1743 from nouiz/repeat

Fix repeat op tests on windows.
...@@ -291,10 +291,10 @@ class TestRepeatOp(utt.InferShapeTester): ...@@ -291,10 +291,10 @@ class TestRepeatOp(utt.InferShapeTester):
self.op = RepeatOp() self.op = RepeatOp()
# uint64 always fails # uint64 always fails
# int64 and uint32 also fail if python int are 32-bit # int64 and uint32 also fail if python int are 32-bit
int_bitwidth = theano.gof.python_int_bitwidth() ptr_bitwidth = theano.gof.local_bitwidth()
if int_bitwidth == 64: if ptr_bitwidth == 64:
self.numpy_unsupported_dtypes = ('uint64',) self.numpy_unsupported_dtypes = ('uint64',)
if int_bitwidth == 32: if ptr_bitwidth == 32:
self.numpy_unsupported_dtypes = ('uint32', 'int64', 'uint64') self.numpy_unsupported_dtypes = ('uint32', 'int64', 'uint64')
def test_repeatOp(self): def test_repeatOp(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论