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);