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

test gpu elemwise fusion only on 3d tensor instead of 4d.

This fix the remaining case error in the buildbot. This don't hide the new case fused in the previous commit as their was error before it in the gpu 3d case. The error in 4d are not real error as they are caused by limit in the number of input in a gpu elemwise op we can put in the gpu code that exist now.
上级 1f016055
......@@ -855,8 +855,8 @@ class test_fusion(unittest.TestCase):
mode._optimizer=mode._optimizer.including('local_elemwise_fusion','canonicalize')
self.do(mode, shared, shp)
def test_elemwise_fusion_4d(self):
shp=(5,5,5,5)
def test_elemwise_fusion_3d(self):
shp=(3,3,3)
mode=copy.copy(compile.mode.get_default_mode())
#we need the optimisation enabled and the canonicalize.
#the canonicalize is needed to merge multiplication/addition by constant.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论