diff --git a/mindspore/ccsrc/kernel/tbe/tbe_adapter.cc b/mindspore/ccsrc/kernel/tbe/tbe_adapter.cc index 406495552644899aaa5f44c25b22f570f9f61d0c..582e40e2df5b01a258d5727022ac6c10c99366fa 100644 --- a/mindspore/ccsrc/kernel/tbe/tbe_adapter.cc +++ b/mindspore/ccsrc/kernel/tbe/tbe_adapter.cc @@ -69,7 +69,7 @@ static std::map tbe_func_adapter_map = { {"reduce_sum", "reduce_sum_d"}, {"one_hot", "one_hot_d"}, {"sum", "reduce_sum_d"}, - {"lamb_next_mv_with_decay_v1", "lamb_next_m_v_with_decay"}, + {"lamb_next_mv_with_decay", "lamb_next_m_v_with_decay"}, {"lamb_next_mv", "lamb_next_m_v"}, {"split", "split_d"}, {"resize_nearest_neighbor", "resize_nearest_neighbor_v2_d"}, diff --git a/mindspore/ccsrc/pre_activate/ascend/ascend_backend_optimization.cc b/mindspore/ccsrc/pre_activate/ascend/ascend_backend_optimization.cc index 0b4027df590c5eaf1e1085f37f52dda43ba7cfa5..afe55aa27f364f816cf886ea8111fa854b5dc501 100644 --- a/mindspore/ccsrc/pre_activate/ascend/ascend_backend_optimization.cc +++ b/mindspore/ccsrc/pre_activate/ascend/ascend_backend_optimization.cc @@ -31,7 +31,6 @@ #include "pre_activate/ascend/ir_fusion/confusion_softmax_grad_rule.h" #include "pre_activate/ascend/ir_fusion/lamb_next_mv_rule.h" #include "pre_activate/ascend/ir_fusion/lamb_next_mv_with_decay_rule.h" -#include "pre_activate/ascend/ir_fusion/lamb_next_mv_with_decay_v1_rule.h" #include "pre_activate/ascend/ir_fusion/lamb_next_right_rule.h" #include "pre_activate/ascend/ir_fusion/lamb_update_with_lr_v2.h" #include "pre_activate/ascend/ir_fusion/layer_norm_beta_gamma_backprop_fusion.h" @@ -82,7 +81,6 @@ void AddAscendBackendOptionalIRFusion(PassManager *ir_fusion_pm) { ir_fusion_pm->AddPass(std::make_shared()); ir_fusion_pm->AddPass(std::make_shared()); ir_fusion_pm->AddPass(std::make_shared()); - ir_fusion_pm->AddPass(std::make_shared()); ir_fusion_pm->AddPass(std::make_shared()); ir_fusion_pm->AddPass(std::make_shared()); ir_fusion_pm->AddPass(std::make_shared());