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

refine black list (#56537)

上级 b8d7f801
......@@ -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",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册