!158 hcom parellel,filter allreduce when link with target allreduce

Merge pull request !158 from 王笑天/development
pull/158/MERGE
王涛 4 years ago committed by Gitee
commit 2339737316

@ -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); if ((node->GetType() != HCOMALLREDUCE && node->GetType() != HVDCALLBACKALLREDUCE)) {
node->GetOpDesc()->SetStreamId(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);
}
} }
} }

Loading…
Cancel
Save