microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-04-20 06:49:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

copy to switch chan

Browse Source
This commit is contained in:
Ubuntu
2026-02-20 18:14:33 +00:00
parent db34b946a8
commit aa1ad02a70
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/tutorials/05-switch-channel/bidir_memory_channel.cu
View File

@@ -95,6 +95,7 @@ void worker(int myRank, int gpuId, const std::string &ipPort) {
log("Rank ", myRank, " (GPU ", gpuId, "): synchronize ...");
auto deviceHandle = switchChannel.deviceHandle();
cudaMemcpyToSymbol(gConstSwitchChan, &deviceHandle, sizeof(deviceHandle));
cudaDeviceSynchronize();
log("Rank ", myRank, " (GPU ", gpuId, "): launch kernel ...");
@@ -109,6 +110,8 @@ void worker(int myRank, int gpuId, const std::string &ipPort) {
comm->bootstrap()->barrier();
log("Rank ", myRank, " (GPU ", gpuId, "): kernel launched ...");
//mscclpp::GpuBuffer buffer(bufferBytes);
//mscclpp::GpuBuffer pktBuffer(pktBufferBytes);
//mscclpp::RegisteredMemory localRegMem = comm.registerMemory(buffer.data(), buffer.bytes(), transport);