提交 f244431b authored 作者: Samira Shabanian's avatar Samira Shabanian

alloc has changed

上级 143896b3
...@@ -6783,7 +6783,7 @@ def local_grad_clip(node): ...@@ -6783,7 +6783,7 @@ def local_grad_clip(node):
@register_stabilize @register_stabilize
@register_specialize @register_specialize
@gof.local_optimizer([T.Alloc]) @gof.local_optimizer([T.Alloc])
def local_merge_allo(cnode): def local_merge_alloc(node):
# This opt takes care of several cases: # This opt takes care of several cases:
# Alloc(Alloc(m, x, 1, 1, 1), x, y, z, w) -> Alloc(m, x, y, z, w) # Alloc(Alloc(m, x, 1, 1, 1), x, y, z, w) -> Alloc(m, x, y, z, w)
# Alloc(Alloc(m, y, 1, 1), x, y, z, w) -> Alloc(m, x, y, z, w) # Alloc(Alloc(m, y, 1, 1), x, y, z, w) -> Alloc(m, x, y, z, w)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论