From c70d79a075590ca82fedb30c9b46eaeb6503905b Mon Sep 17 00:00:00 2001
From: Jiabin Yang <360788950@qq.com>
Date: Wed, 24 Aug 2022 18:26:38 +0800
Subject: [PATCH] fix argsort grad not fill zero (#45371)

---
 .../auto_code_generator/final_state_generator/codegen_utils.py | 3 ++-
 1 file changed, 2 insertions(+), 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 64f2602c441..56095963d49 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,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
-- 
GitLab