From 9ba2dc80f0f131a13df3a5de84c87b9582065dd6 Mon Sep 17 00:00:00 2001 From: wangxiaotian22 Date: Thu, 29 Oct 2020 20:58:32 +0800 Subject: [PATCH] hcom parellel,filter allreduce when link with target allreduce --- ge/graph/build/logical_stream_allocator.cc | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) 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); + } } }