Update EnvVars.cpp

[ROCm/rccl commit: 757d1891e9]
2024-09-04 16:55:36 -04:00
parent dc04844405
commit 1bd3db8fc7
1 changed files with 1 additions and 0 deletions
@@ -94,6 +94,7 @@ namespace RcclUnitTesting
    // NOTE: Cannot use HIP call prior to launching unless it is inside another child process
    numDetectedGpus = 0;
    getDeviceCount(&numDetectedGpus);
+    numDetectedGpus = min(numDetectedGpus, 16);
    isGfx94 = false;
    getArchInfo(&isGfx94, "gfx94");
    isGfx12 = false;