提交 191db4ae authored 作者: Frederic Bastien's avatar Frederic Bastien

adde test that check that their is no rebroadcast op in the GpuJoin.

上级 f5d82db8
...@@ -574,6 +574,12 @@ def test_gpujoin_assert_cndas(): ...@@ -574,6 +574,12 @@ def test_gpujoin_assert_cndas():
assert False assert False
def test_gpujoin_no_rebroadcast():
_a = numpy.asarray([[1,2],[3,4]],dtype='float32')
a = tcn.shared_constructor(_a)
f = theano.function([],T.join(1,a))
l = f.maker.env.toposort()
assert not any([isinstance(x.op,T.Rebroadcast) for x in l])
if __name__ == '__main__': if __name__ == '__main__':
test_gpujoin_twomatrices_joincolumns() test_gpujoin_twomatrices_joincolumns()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论