diff --git a/ge/graph/build/logical_stream_allocator.cc b/ge/graph/build/logical_stream_allocator.cc index 5b8ce824..72337b26 100644 --- a/ge/graph/build/logical_stream_allocator.cc +++ b/ge/graph/build/logical_stream_allocator.cc @@ -507,8 +507,10 @@ Status AllReduceParallelPass::Run(ComputeGraphPtr graph, const vectorGetName().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); + } } }