microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-11 08:50:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

address hang issue

Browse Source
This commit is contained in:
Binyang Li
2026-05-09 03:16:33 +00:00
parent e208cc326b
commit 825fc124a5
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/ext/collectives/allreduce/allreduce_nvls_packet.cu
View File

@@ -83,6 +83,7 @@ void AllreduceNvlsPacket::initialize(std::shared_ptr<Communicator> comm) {
this->nvlsConnections_ = setupNvlsConnections(comm, nvlsBufferSize_, nSwitchChannels);
this->switchChannels_ =
setupNvlsChannels(this->nvlsConnections_, this->scratchBuffer_, this->scratchBufferSize_, nSwitchChannels);
comm->bootstrap()->barrier();
}
AlgorithmCtxKey AllreduceNvlsPacket::generateAllreduceContextKey(const void*, void*, size_t, DataType, bool) {