Add side effect flag for push kernel to bypass cse.

2020-09-08 19:59:37 +08:00 · 2020-09-08 19:59:37 +08:00 · 78f6478a57
parent 65c5663c8f
commit 78f6478a57
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):