Fix MPI test usage

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

1 files changed, 3 insertions, 1 deletions

diff --git a/test/mpi/mpi_test.cu b/test/mpi/mpi_test.cu
index 54ebbce..ee86edc 100644
--- a/test/mpi/mpi_test.cu
+++ b/test/mpi/mpi_test.cu
@@ -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]);