fix argsort grad not fill zero (#45371)

c70d79a0 · Jiabin Yang · GitHub · 1badb4c6 · c70d79a0
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

paddle/fluid/eager/auto_code_generator/final_state_generator/codegen_utils.py ...uto_code_generator/final_state_generator/codegen_utils.py +2 -1

未找到文件。
--- 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,8 @@ 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", "expand_grad"
+    "depthwise_conv2d_grad_grad", "concat_double_grad", "expand_grad",
+    "argsort_grad"
 ])
 # For API dispatch used at python-level