From b71e8ffc0b1202ae4af22e57011c21a9a96e24af Mon Sep 17 00:00:00 2001
From: wanghuancoder <wanghuan29@baidu.com>
Date: Wed, 23 Aug 2023 09:30:18 +0800
Subject: [PATCH] refine black list (#56537)

---
 .../new_executor/new_ir_interpreter.cc        | 26 +++++++++----------
 1 file changed, 13 insertions(+), 13 deletions(-)

diff --git a/paddle/fluid/framework/new_executor/new_ir_interpreter.cc b/paddle/fluid/framework/new_executor/new_ir_interpreter.cc
index 9587511af77..28752ef1f7e 100644
--- a/paddle/fluid/framework/new_executor/new_ir_interpreter.cc
+++ b/paddle/fluid/framework/new_executor/new_ir_interpreter.cc
@@ -346,9 +346,6 @@ void NewIRInterpreter::UpdateSyncOpNum() {
 
 void NewIRInterpreter::UpdateNcclOpNum() {
   static std::set<std::string> nccl_op_set = {
-      "pd.sync_batch_norm_",
-      "pd.sync_batch_norm",
-      "pd.sync_batch_norm_grad",
       "pd.c_softmax_with_cross_entropy",
       "pd.c_allgather",
       "pd.c_allreduce_max",
@@ -360,6 +357,7 @@ void NewIRInterpreter::UpdateNcclOpNum() {
       "pd.c_reduce_prod",
       "pd.c_reducescatter",
       "pd.c_broadcast",
+      "pd.c_broadcast_",
       "pd.c_scatter",
       "pd.partial_send",
       "pd.partial_recv",
@@ -370,11 +368,12 @@ void NewIRInterpreter::UpdateNcclOpNum() {
       "pd.barrier",
       "pd.alltoall",
       "pd.global_gather",
-      "pd.distributed_fused_lamb_op",
-      "pd.margin_cross_entropy_op",
-      "pd.sync_batch_norm_op",
-      "pd.data_norm_op",
-      "pd.class_center_sample_op",
+      "pd.distributed_fused_lamb",
+      "pd.margin_cross_entropy",
+      "pd.sync_batch_norm",
+      "pd.sync_batch_norm_",
+      "pd.data_norm",
+      "pd.class_center_sample",
       "pd.all_to_all",
       "pd.dist_concat",
       "pd.all_gather",
@@ -405,11 +404,12 @@ void NewIRInterpreter::UpdateNcclOpNum() {
       "pd.barrier_grad",
       "pd.alltoall_grad",
       "pd.global_gather_grad",
-      "pd.distributed_fused_lamb_op_grad",
-      "pd.margin_cross_entropy_op_grad",
-      "pd.sync_batch_norm_op_grad",
-      "pd.data_norm_op_grad",
-      "pd.class_center_sample_op_grad",
+      "pd.distributed_fused_lamb_grad",
+      "pd.margin_cross_entropy_grad",
+      "pd.margin_cross_entropy_grad_"
+      "pd.sync_batch_norm_grad",
+      "pd.data_norm_grad",
+      "pd.class_center_sample_grad",
       "pd.all_to_all_grad",
       "pd.dist_concat_grad",
       "pd.all_gather_grad",
-- 
GitLab