modify ir_grad (#57449)

PaddlePaddle · Sep 19, 2023 · f331c38 · f331c38
1 parent a79ba09
commit f331c38
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/paddle/autograd/ir_backward.py b/python/paddle/autograd/ir_backward.py
@@ -528,8 +528,8 @@ def update_input_grad_map(op, input_grads):
                         ] = state.value_to_valuegrad[value]
                     else:
                         state.op_to_opgrad[op] = []
-                else:
-                    state.op_to_opgrad[op] = []
+            else:
+                state.op_to_opgrad[op] = []
 
 
 def create_backward_prune_set(inputs, outputs, no_grad_set, state):