Revert "fixing the error message for mpirun when number of requested GPUs exceeds the limits (#33)" (#36)
This reverts commit e146460810.
This commit is contained in:
committed by
GitHub
orang tua
e146460810
melakukan
e856fa720f
@@ -1127,13 +1127,11 @@ int main(int argc, char* argv[]) {
|
||||
}
|
||||
|
||||
HIPCHECK(hipGetDeviceCount(&numDevices));
|
||||
#ifndef MPI_SUPPORT
|
||||
if (nGpus > numDevices)
|
||||
{
|
||||
fprintf(stderr, "[ERROR] The number of requested GPUs (%d) is greater than the number of GPUs available (%d)\n", nGpus, numDevices);
|
||||
return testNcclError;
|
||||
}
|
||||
#endif
|
||||
if (minBytes > maxBytes) {
|
||||
fprintf(stderr, "invalid sizes for 'minbytes' and 'maxbytes': %llu > %llu\n",
|
||||
(unsigned long long)minBytes,
|
||||
@@ -1156,14 +1154,7 @@ int main(int argc, char* argv[]) {
|
||||
return -1;
|
||||
}
|
||||
#ifdef MPI_SUPPORT
|
||||
int nProcs = 1;
|
||||
MPI_Init(&argc, &argv);
|
||||
MPI_Comm_size(MPI_COMM_WORLD, &nProcs);
|
||||
if (nGpus * nProcs > numDevices)
|
||||
{
|
||||
fprintf(stderr, "[ERROR] The number of requested GPUs (%d) is greater than the number of GPUs available (%d)\n", nGpus*nProcs, numDevices);
|
||||
return testNcclError;
|
||||
}
|
||||
#endif
|
||||
TESTCHECK(run());
|
||||
return 0;
|
||||
|
||||
Reference in New Issue
Block a user