mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

not process non task split when kernel by kernel

This commit is contained in:
reku1997 2022-09-20 19:13:20 +08:00
parent c1efb947ca
commit cdb874a942
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/plugin/device/ascend/hal/hardware/ascend_somas.cc
View File

@ -194,6 +194,12 @@ void AscendSomas::IndependentNodeOutputProcess(const session::KernelGraph &graph
}
void AscendSomas::NonTaskSplitProcess(const session::KernelGraph &graph) {
// When not used task sink mode, should not process non task split mem-reuse.
// Because the logic for processing non task split memory offset is in the function TaskGenerator:: GetTaskInput,
// only run in task sink mode.
if (!graph.is_graph_run_mode()) {
return;
}
auto &kernel_cnodes = graph.execution_order();
for (const auto &kernel : kernel_cnodes) {
auto op_name = common::AnfAlgo::GetCNodeName(kernel);