fix filter parameter in to_static

PaddlePaddle · Sep 27, 2023 · f584c2f · f584c2f
1 parent 0b70e22
commit f584c2f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/paddle/fluid/eager/to_static/run_program_op_func.h b/paddle/fluid/eager/to_static/run_program_op_func.h
@@ -291,9 +291,7 @@ inline void newir_run_program_ad_func(
     // Set TensorWrappers
     grad_node->SetFwdX(filter_x);
 
-    auto filter_params =
-        newir_filter_unused_input_var_in_backward(params, "bp", attrs);
-    grad_node->SetFwdParams(filter_params);
+    grad_node->SetFwdParams(params);
 
     grad_node->SetStepScope(step_scope);  // just for set useable.
 

diff --git a/python/paddle/decomposition/rules.py b/python/paddle/decomposition/rules.py
@@ -178,7 +178,7 @@ def dropout_composite(x, seed_tensor, p, is_test, mode, seed, fix_seed):
                     shape=x.shape, value=(1.0 - p), dtype=x.dtype
                 ), cast(mask, uint8_type)
         else:
-            return x, cast(
+            return x * 1.0, cast(
                 mask, uint8_type
             )  # assign(x), cast(mask, mask, core.VarDesc.VarType.UINT8)
     else: