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 7dbc93a35817d086aa3f1f81fd2f310b568c1363..c45f751b8c3dc94d3509a2d62e228f781ff5b356 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", "concat_double_grad" + "depthwise_conv2d_grad_grad", "concat_double_grad", "expand_grad" ]) # For API dispatch used at python-level