Browse Source

enable TensorScatterUpdateFission pass in pynative mode

tags/v0.6.0-beta
huanghui 5 years ago
parent
commit
a956ec1e85
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/ccsrc/pre_activate/ascend/ascend_backend_optimization.cc

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

@@ -292,6 +292,7 @@ void RunOpAscendBackendIRFusionOptimization(const std::shared_ptr<session::Kerne
ir_fusion_pm->AddPass(std::make_shared<TopKSplit>()); ir_fusion_pm->AddPass(std::make_shared<TopKSplit>());
ir_fusion_pm->AddPass(std::make_shared<AddnFission>()); ir_fusion_pm->AddPass(std::make_shared<AddnFission>());
ir_fusion_pm->AddPass(std::make_shared<InsertPadForNMSWithMask>()); ir_fusion_pm->AddPass(std::make_shared<InsertPadForNMSWithMask>());
ir_fusion_pm->AddPass(std::make_shared<TensorScatterUpdateFission>());


optimizer->AddPassManager(ir_fusion_pm); optimizer->AddPassManager(ir_fusion_pm);
(void)optimizer->Optimize(kernel_graph); (void)optimizer->Optimize(kernel_graph);


Loading…
Cancel
Save