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]
Tá an tiomantas seo le fáil i:
Wenkai Du údaraithe agus GitHub tiomanta 2021-11-16 10:28:26 -08:00
1 tuismitheoir 5bc7a6b858
tiomantas a14a57ea71
14 comhad athraithe
+30 -15

Níor tháinig aon chomhaid le do chuardach

+1 -1
Féach ar an gComhad
@@ -59,7 +59,7 @@ namespace CorrectnessTests
// Number of elements
testing::Values(1024, 1048576),
// 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("")),