From a956ec1e85a3322d9547fb56e05cd75731e031a6 Mon Sep 17 00:00:00 2001 From: huanghui Date: Mon, 13 Jul 2020 11:57:14 +0800 Subject: [PATCH] enable TensorScatterUpdateFission pass in pynative mode --- .../ccsrc/pre_activate/ascend/ascend_backend_optimization.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/mindspore/ccsrc/pre_activate/ascend/ascend_backend_optimization.cc b/mindspore/ccsrc/pre_activate/ascend/ascend_backend_optimization.cc index 01387b15da..48ce87629c 100644 --- a/mindspore/ccsrc/pre_activate/ascend/ascend_backend_optimization.cc +++ b/mindspore/ccsrc/pre_activate/ascend/ascend_backend_optimization.cc @@ -292,6 +292,7 @@ void RunOpAscendBackendIRFusionOptimization(const std::shared_ptrAddPass(std::make_shared()); ir_fusion_pm->AddPass(std::make_shared()); ir_fusion_pm->AddPass(std::make_shared()); + ir_fusion_pm->AddPass(std::make_shared()); optimizer->AddPassManager(ir_fusion_pm); (void)optimizer->Optimize(kernel_graph);