From 8761982986db2e382382637902a87c82b6213d48 Mon Sep 17 00:00:00 2001 From: Weilong Wu Date: Wed, 13 Jul 2022 11:12:54 +0800 Subject: [PATCH] [Eager] concat_double_grad fill zero for empty grads (#44252) --- .../auto_code_generator/final_state_generator/codegen_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/eager/auto_code_generator/final_state_generator/codegen_utils.py b/paddle/fluid/eager/auto_code_generator/final_state_generator/codegen_utils.py index dee3b3d79a2..cd5805740be 100644 --- a/paddle/fluid/eager/auto_code_generator/final_state_generator/codegen_utils.py +++ b/paddle/fluid/eager/auto_code_generator/final_state_generator/codegen_utils.py @@ -32,7 +32,7 @@ ops_to_fill_zero_for_empty_grads = set([ "square_double_grad", "celu_double_grad", "pad_double_grad", "pad3d_double_grad", "squeeze_double_grad", "unsqueeze_double_grad", "instance_norm_double_grad", "conv3d_double_grad", - "depthwise_conv2d_grad_grad" + "depthwise_conv2d_grad_grad", "concat_double_grad" ]) # For API dispatch used at python-level -- GitLab