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

Merge pull request !5917 from ZPaC/master
5 years ago · fc5a7bc9c0
parent 8a20b5d784 78f6478a57
commit fc5a7bc9c0
1 changed files with 1 additions and 0 deletions
--- a/mindspore/ops/operations/other_ops.py
+++ b/mindspore/ops/operations/other_ops.py
@ -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):