Browse Source

add split pass on pynative mode

tags/v0.7.0-beta
jjfeing 5 years ago
parent
commit
b86b5a836a
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/ccsrc/backend/optimizer/ascend/ascend_backend_optimization.cc

+ 1
- 0
mindspore/ccsrc/backend/optimizer/ascend/ascend_backend_optimization.cc View File

@@ -308,6 +308,7 @@ void RunOpAscendBackendIRFusionOptimization(const std::shared_ptr<session::Kerne
}
auto optimizer = std::make_shared<GraphOptimizer>();
auto ir_fusion_pm = std::make_shared<PassManager>("ir_fusion_pm");
ir_fusion_pm->AddPass(std::make_shared<SplitFission>());
ir_fusion_pm->AddPass(std::make_shared<BnSplit>());
ir_fusion_pm->AddPass(std::make_shared<LayerNormGradSplit>());
ir_fusion_pm->AddPass(std::make_shared<TopKSplit>());


Loading…
Cancel
Save