gtest: dynamically generate tests based on test machine's GPU count (#467)

* gtest: dynamically generate tests based on test machine's GPU count * Adjust test element size and bfloat16 threshold for up to 16 GPUs [ROCm/rccl commit: 03a830293c]
2021-11-16 10:28:26 -08:00
parent 5bc7a6b858
commit a14a57ea71
14 changed files with 30 additions and 15 deletions
@@ -67,7 +67,7 @@ namespace CorrectnessTests
                                // Number of elements
                                testing::Values(2520, 3026520),
                                // Number of devices
-                                testing::Values(2,3,4,5,6,7,8),
+                                testing::Range(2,(GTESTS_NUM_GPUS+1)),
                                // In-place or not
                                testing::Values(false),
                                testing::Values("")),