提交 ebf5faf6 authored 作者: Shawn Tan's avatar Shawn Tan

Remove optimisation for `op.set_instead_of_inc=True`

上级 c80355fc
...@@ -1097,7 +1097,10 @@ def local_gpua_advanced_incsubtensor1(op, context_name, inputs, outputs): ...@@ -1097,7 +1097,10 @@ def local_gpua_advanced_incsubtensor1(op, context_name, inputs, outputs):
@op_lifter([tensor.AdvancedIncSubtensor]) @op_lifter([tensor.AdvancedIncSubtensor])
@register_opt2([tensor.AdvancedIncSubtensor], 'fast_compile') @register_opt2([tensor.AdvancedIncSubtensor], 'fast_compile')
def local_gpua_advanced_incsubtensor(op, context_name, inputs, outputs): def local_gpua_advanced_incsubtensor(op, context_name, inputs, outputs):
return GpuAdvancedIncSubtensor() if not op.set_instead_of_inc:
return GpuAdvancedIncSubtensor()
else:
return False
@register_inplace() @register_inplace()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论