提交 5e70dc3e authored 作者: Saizheng Zhang's avatar Saizheng Zhang

solve confilcts in test_opt.py, test_subtensor_inc_subtensor

上级 d30b8fca
...@@ -1895,7 +1895,7 @@ def local_track_shape_i(node): ...@@ -1895,7 +1895,7 @@ def local_track_shape_i(node):
return [shape_feature.shape_of[replacement][node.op.i]] return [shape_feature.shape_of[replacement][node.op.i]]
# TH-2801 opt: subtensor(incsubtensor) # Subtensor(SetSubtensor(x, y, idx), idx) -> y
@register_specialize @register_specialize
@register_canonicalize @register_canonicalize
@gof.local_optimizer([Subtensor]) @gof.local_optimizer([Subtensor])
......
...@@ -6768,6 +6768,7 @@ if __name__ == '__main__': ...@@ -6768,6 +6768,7 @@ if __name__ == '__main__':
t.setUp() t.setUp()
# t.test_perform() # t.test_perform()
t.test_infer_shape() t.test_infer_shape()
test_subtensor_inc_subtensor()
""" """
# unittest.main() # unittest.main()
test_fusion().tes_memory_leak() test_fusion().tes_memory_leak()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论