From adc5857eb298572ddeba46928853bc2a05afefa2 Mon Sep 17 00:00:00 2001 From: wangxiaotian22 Date: Tue, 27 Oct 2020 21:08:32 +0800 Subject: [PATCH] hcom parellel,filter allreduce when link with target allreduce --- src/ge/graph/build/logical_stream_allocator.cc | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/ge/graph/build/logical_stream_allocator.cc b/src/ge/graph/build/logical_stream_allocator.cc index d1866584..00b3640f 100644 --- a/src/ge/graph/build/logical_stream_allocator.cc +++ b/src/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); + } } }