diff --git a/src/common.cu b/src/common.cu index def6ca029e..691c2402dd 100644 --- a/src/common.cu +++ b/src/common.cu @@ -1100,7 +1100,7 @@ testResult_t run() { for (int p=0; p numDevices) + if (nGpus * localSize > numDevices && numDevices != 1) { fprintf(stderr, "[ERROR] The number of requested GPUs (%d) is greater than the number of GPUs available (%d) on node (%s)\n", nGpus*localSize, numDevices, hostname); return testNcclError; @@ -1124,7 +1124,7 @@ testResult_t run() { char* envstr = getenv("NCCL_TESTS_DEVICE"); int gpu0 = envstr ? atoi(envstr) : -1; for (int i=0; i