Browse Source

Pre Merge pull request !157 from 王笑天/master

pull/157/MERGE
王笑天 Gitee 4 years ago
parent
commit
919b6b040e
1 changed files with 4 additions and 2 deletions
  1. +4
    -2
      src/ge/graph/build/logical_stream_allocator.cc

+ 4
- 2
src/ge/graph/build/logical_stream_allocator.cc View File

@@ -507,8 +507,10 @@ Status AllReduceParallelPass::Run(ComputeGraphPtr graph, const vector<SubgraphPt
old_stream_to_new.emplace(old_stream, new_stream); old_stream_to_new.emplace(old_stream, new_stream);
} }


GELOGI("Stream of node %s has been updated from %ld to %ld.", node->GetName().c_str(), old_stream, new_stream);
node->GetOpDesc()->SetStreamId(new_stream);
if ((node->GetType() != HCOMALLREDUCE && node->GetType() != HVDCALLBACKALLREDUCE)) {
GELOGI("Stream of node %s has been updated from %ld to %ld.", node->GetName().c_str(), old_stream, new_stream);
node->GetOpDesc()->SetStreamId(new_stream);
}
} }
} }




Loading…
Cancel
Save