microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-04 05:31:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Align interfaces of put/get/putPackets/getPackets (#185)

Browse Source
This commit is contained in:
Changho Hwang
2023-10-07 22:18:26 +08:00
committed by GitHub
parent 497a9e0c82
commit 11ac824cc7
7 changed files with 93 additions and 81 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/test/simple_proxy_channel_test.cu
View File

@@ -26,7 +26,7 @@ extern "C" __global__ void __launch_bounds__(1024, 1)
channels[tid].put(2 * my_offset, 2 * my_offset, 2 * size_per_rank);
}
if (my_nghr != my_rank && my_nghr < nranks)
mscclpp::getPackets(data, my_nghr_offset, scratch, 2 * my_nghr_offset, size_per_rank, tid % nthreads_per_rank,
mscclpp::getPackets(scratch, 2 * my_nghr_offset, data, my_nghr_offset, size_per_rank, tid % nthreads_per_rank,
nthreads_per_rank, flag);
} else {
if (tid < nranks && tid != my_rank) {