Fix MPI test usage

Only display usage from rank 0 and exit instead of continuing (and seg fault).


[ROCm/rccl commit: 9de361a1b9]
This commit is contained in:
Sylvain Jeaugey
2016-04-19 10:43:38 -07:00
parent 014160c90a
commit 6e57ed83df
+3 -1
View File
@@ -47,7 +47,9 @@ int main(int argc, char *argv[]) {
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
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]);