Browse Source

hcom parellel,filter allreduce when link with target allreduce

tags/v1.1.0
wangxiaotian22 4 years ago
parent
commit
9ba2dc80f0
1 changed files with 4 additions and 2 deletions
  1. +4
    -2
      ge/graph/build/logical_stream_allocator.cc

+ 4
- 2
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);
}

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