提交 fc5a7bc9 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!5917 Add side effect flag for push kernel to bypass cse.

Merge pull request !5917 from ZPaC/master
......@@ -532,6 +532,7 @@ class Push(PrimitiveWithInfer):
def __init__(self, optim_type='ApplyMomentum', only_shape_indices=None):
"""init Push"""
self.add_prim_attr("primitive_target", "CPU")
self.add_prim_attr("_side_effect", True)
self.init_prim_io_names(inputs=['optim_inputs', 'optim_input_shapes'], outputs=['key'])
def infer_shape(self, inputs, shapes):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册