diff --git a/projects/rccl/test/mpi/mpi_test.cu b/projects/rccl/test/mpi/mpi_test.cu index 54ebbce995..ee86edcb3b 100644 --- a/projects/rccl/test/mpi/mpi_test.cu +++ b/projects/rccl/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 \n", argv[0]); + if (rank == 0) + printf("Usage : %s \n", argv[0]); + exit(1); } int gpu = atoi(argv[rank+1]);