Browse Source

remove old ir fusion pass lamb next mv with decay v1

tags/v0.3.0-alpha
zhaozhenlong 5 years ago
parent
commit
9d34b31ee9
2 changed files with 1 additions and 3 deletions
  1. +1
    -1
      mindspore/ccsrc/kernel/tbe/tbe_adapter.cc
  2. +0
    -2
      mindspore/ccsrc/pre_activate/ascend/ascend_backend_optimization.cc

+ 1
- 1
mindspore/ccsrc/kernel/tbe/tbe_adapter.cc View File

@@ -69,7 +69,7 @@ static std::map<string, string> 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"},


+ 0
- 2
mindspore/ccsrc/pre_activate/ascend/ascend_backend_optimization.cc View File

@@ -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<ClipByNormNoDivSquareSumFusion>());
ir_fusion_pm->AddPass(std::make_shared<LambUpdateWithLRRuleFusion>());
ir_fusion_pm->AddPass(std::make_shared<ConfusionSoftmaxGradRule>());
ir_fusion_pm->AddPass(std::make_shared<LambNextMVWithDecayV1Rule>());
ir_fusion_pm->AddPass(std::make_shared<LambNextMVRule>());
ir_fusion_pm->AddPass(std::make_shared<LambNextMVWithDecayRule>());
ir_fusion_pm->AddPass(std::make_shared<LambNextRightRule>());


Loading…
Cancel
Save