yezhengmao/nccl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix MPI test usage

Browse Source 
Only display usage from rank 0 and exit instead of continuing (and seg fault).
This commit is contained in:
Sylvain Jeaugey 2016-04-19 10:43:38 -07:00
parent b16cc5d197
commit 9de361a1b9
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test/mpi/mpi_test.cu
View File

@ -47,7 +47,9 @@ int main(int argc, char *argv[]) {
MPI_Comm_rank(MPI_COMM_WORLD, &rank); MPI_Comm_rank(MPI_COMM_WORLD, &rank);
if (argc < size) { if (argc < size) {
printf("Usage : %s <GPU list per rank>\n", argv[0]); if (rank == 0)
printf("Usage : %s <GPU list per rank>\n", argv[0]);
exit(1);
} }
int gpu = atoi(argv[rank+1]); int gpu = atoi(argv[rank+1]);