Checking that number of requested GPUs is not more than number of available GPUs
[ROCm/rccl-tests commit: 043eef6999]
Ta zmiana jest zawarta w:
%! (template.HTML=
Gilbert Lee)złożyło 2019-07-11 15:36:21 +00:00
1 rodzic
30d348de0e
złożenie
0288eeef75
Plik 1 zmieniony
+9
@@ -738,6 +738,15 @@ int main(int argc, char* argv[]) {
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
|
||||
int numDevices;
|
||||
HIPCHECK(hipGetDeviceCount(&numDevices));
|
||||
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;
|
||||
}
|
||||
|
||||
#ifdef MPI_SUPPORT
|
||||
MPI_Init(&argc, &argv);
|
||||
#endif
|
||||
|
||||
Reference in new issue
Zablokuj użytkownika