From 314c0cbdb256fb1e041a50ce30cf2cbea800d8c1 Mon Sep 17 00:00:00 2001 From: liuxiao93 Date: Mon, 18 Jan 2021 19:15:39 +0800 Subject: [PATCH] split optimizer --- .../backend/optimizer/ascend/enhancer/split_n_optimizer.cc | 2 +- .../ccsrc/runtime/device/ascend/tasksink/task_generator.cc | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/mindspore/ccsrc/backend/optimizer/ascend/enhancer/split_n_optimizer.cc b/mindspore/ccsrc/backend/optimizer/ascend/enhancer/split_n_optimizer.cc index 94ce8cbf0b..ef163e9e0c 100644 --- a/mindspore/ccsrc/backend/optimizer/ascend/enhancer/split_n_optimizer.cc +++ b/mindspore/ccsrc/backend/optimizer/ascend/enhancer/split_n_optimizer.cc @@ -135,7 +135,7 @@ bool OutputCheck(const FuncGraphPtr &func_graph, const AnfNodePtr &node) { MS_LOG(INFO) << "Next node is not a AICore node, can not optimizer."; return false; } - if (func_graph->output() == item || AnfAlgo::CheckPrimitiveType(node, prim::kPrimReturn)) { + if (func_graph->output() == item || AnfAlgo::CheckPrimitiveType(item, prim::kPrimReturn)) { MS_LOG(INFO) << "Next node is graph output or return, can not optimizer."; return false; } diff --git a/mindspore/ccsrc/runtime/device/ascend/tasksink/task_generator.cc b/mindspore/ccsrc/runtime/device/ascend/tasksink/task_generator.cc index c31dbe2fa0..922c4b10d6 100644 --- a/mindspore/ccsrc/runtime/device/ascend/tasksink/task_generator.cc +++ b/mindspore/ccsrc/runtime/device/ascend/tasksink/task_generator.cc @@ -171,8 +171,7 @@ bool TaskGenerator::LaunchKernel(const CNodePtr &anf_node_ptr, uint32_t stream_i input->addr = static_cast(split_input0_device_address->ptr_) + (prenode_with_index.second * input->size); MS_LOG(INFO) << "Change " << anf_node_ptr->fullname_with_scope() << "'s input " << i << " address to " - << split_input0_device_address->ptr_ << " + " - << "prenode_with_index.second * input->size"; + << split_input0_device_address->ptr_ << " + " << prenode_with_index.second * input->size; } } kernel_inputs.push_back(input);