microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-13 17:55:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

more clean up

Browse Source
This commit is contained in:
Saeed Maleki
2023-02-20 00:23:24 +00:00
parent 4f3418aa77
commit ca89c17aaa
1 changed files with 15 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
tests/p2p_test_mpi.cu
View File

@@ -4,6 +4,15 @@
#include <stdlib.h>
#include <unistd.h>
#define MSCCLPPCHECK(call) do { \
mscclppResult_t res = call; \
if (res != mscclppSuccess && res != mscclppInProgress) { \
/* Print the back trace*/ \
printf("Failure at %s:%d -> %d", __FILE__, __LINE__, res); \
return res; \
} \
} while (0);
// Check CUDA RT calls
#define CUDACHECK(cmd) do { \
cudaError_t err = cmd; \
@@ -82,42 +91,20 @@ int main(int argc, const char *argv[])
mscclppResult_t res;
// if (rank == 0)
// sleep(10);
// else
// sleep(10);
mscclppDevConn_t devConns[8];
// Read from all other ranks
for (int r = 0; r < world_size; ++r) {
if (r == rank) continue;
int tag = 0;
res = mscclppConnect(comm, &devConns[r], r, data_d, flag_d, tag, mscclppTransportP2P);
if (res != mscclppSuccess) {
printf("mscclppConnect failed\n");
return -1;
}
}
// Let others read from me
// for (int r = 0; r < world_size; ++r) {
// if (r == rank) continue;
// int tag = r * world_size + rank;
// res = mscclppConnect(comm, r, rank, data_d, flag_d, tag, mscclppTransportP2P);
// if (res != mscclppSuccess) {
// printf("mscclppConnect failed\n");
// return -1;
// }
// }
res = mscclppConnectionSetup(comm);
if (res != mscclppSuccess) {
printf("mscclppConnectionSetup failed\n");
return -1;
MSCCLPPCHECK(mscclppConnect(comm, &devConns[r], r, data_d, flag_d, tag, mscclppTransportP2P));
}
MSCCLPPCHECK(mscclppConnectionSetup(comm));
CUDACHECK(cudaMemcpyToSymbol(constDevConns, devConns, sizeof(mscclppDevConn_t) * world_size));
kernel<<<1, 1>>>(rank, world_size);
CUDACHECK(cudaDeviceSynchronize());
int *buf = (int *)calloc(world_size, sizeof(int));
@@ -134,11 +121,7 @@ int main(int argc, const char *argv[])
}
}
res = mscclppCommDestroy(comm);
if (res != mscclppSuccess) {
printf("mscclppDestroy failed\n");
return -1;
}
MSCCLPPCHECK(mscclppCommDestroy(comm));
MPI_Finalize();