Merge pull request #5 from 2742195759/pr_48213

[bugfix] fix _find_op_path_ bugs in append_backward.
2742195759 · Nov 24, 2022 · f380f10 · f380f10
2 parents 53b0eb6 + 25968b4
commit f380f10
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/python/paddle/fluid/backward.py b/python/paddle/fluid/backward.py
@@ -2221,6 +2221,10 @@ def _find_op_path_(
                 op.desc.output_arg_names(), output_names
             ):
                 relevant_op_flags[i] = True
+                if core.has_non_empty_grad_op_maker(op.type):
+                    for name in op.desc.input_arg_names():
+                        if name not in no_grad_set:
+                            output_names.add(name)
 
     op_path = [
         block.ops[i] for i in range(len(block.ops)) if relevant_op_flags[i]

diff --git a/python/paddle/fluid/layers/rnn.py b/python/paddle/fluid/layers/rnn.py
@@ -694,7 +694,6 @@ def _switch_grad(x, stop=False):
 
     out_array = control_flow.create_array(dtype=flatten(inputs)[0].dtype)
 
-    init_array = control_flow.create_array(dtype=flatten(inputs)[0].dtype)
     init_array = map_structure(
         lambda x: control_flow.create_array(dtype=x.dtype), initial_states
     )