未验证 提交 b71e8ffc 编写于 作者: W wanghuancoder 提交者: GitHub

refine black list (#56537)

上级 b8d7f801
...@@ -346,9 +346,6 @@ void NewIRInterpreter::UpdateSyncOpNum() { ...@@ -346,9 +346,6 @@ void NewIRInterpreter::UpdateSyncOpNum() {
void NewIRInterpreter::UpdateNcclOpNum() { void NewIRInterpreter::UpdateNcclOpNum() {
static std::set<std::string> nccl_op_set = { 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_softmax_with_cross_entropy",
"pd.c_allgather", "pd.c_allgather",
"pd.c_allreduce_max", "pd.c_allreduce_max",
...@@ -360,6 +357,7 @@ void NewIRInterpreter::UpdateNcclOpNum() { ...@@ -360,6 +357,7 @@ void NewIRInterpreter::UpdateNcclOpNum() {
"pd.c_reduce_prod", "pd.c_reduce_prod",
"pd.c_reducescatter", "pd.c_reducescatter",
"pd.c_broadcast", "pd.c_broadcast",
"pd.c_broadcast_",
"pd.c_scatter", "pd.c_scatter",
"pd.partial_send", "pd.partial_send",
"pd.partial_recv", "pd.partial_recv",
...@@ -370,11 +368,12 @@ void NewIRInterpreter::UpdateNcclOpNum() { ...@@ -370,11 +368,12 @@ void NewIRInterpreter::UpdateNcclOpNum() {
"pd.barrier", "pd.barrier",
"pd.alltoall", "pd.alltoall",
"pd.global_gather", "pd.global_gather",
"pd.distributed_fused_lamb_op", "pd.distributed_fused_lamb",
"pd.margin_cross_entropy_op", "pd.margin_cross_entropy",
"pd.sync_batch_norm_op", "pd.sync_batch_norm",
"pd.data_norm_op", "pd.sync_batch_norm_",
"pd.class_center_sample_op", "pd.data_norm",
"pd.class_center_sample",
"pd.all_to_all", "pd.all_to_all",
"pd.dist_concat", "pd.dist_concat",
"pd.all_gather", "pd.all_gather",
...@@ -405,11 +404,12 @@ void NewIRInterpreter::UpdateNcclOpNum() { ...@@ -405,11 +404,12 @@ void NewIRInterpreter::UpdateNcclOpNum() {
"pd.barrier_grad", "pd.barrier_grad",
"pd.alltoall_grad", "pd.alltoall_grad",
"pd.global_gather_grad", "pd.global_gather_grad",
"pd.distributed_fused_lamb_op_grad", "pd.distributed_fused_lamb_grad",
"pd.margin_cross_entropy_op_grad", "pd.margin_cross_entropy_grad",
"pd.sync_batch_norm_op_grad", "pd.margin_cross_entropy_grad_"
"pd.data_norm_op_grad", "pd.sync_batch_norm_grad",
"pd.class_center_sample_op_grad", "pd.data_norm_grad",
"pd.class_center_sample_grad",
"pd.all_to_all_grad", "pd.all_to_all_grad",
"pd.dist_concat_grad", "pd.dist_concat_grad",
"pd.all_gather_grad", "pd.all_gather_grad",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册