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]
이 커밋은 다음에 포함됨:
Wenkai Du 작성하고 GitHub가 2021-11-16 10:28:26 -08:00를 커밋함
1 부모 5bc7a6b858
커밋 a14a57ea71
14 파일이 변경됨
+30 -15

검색과 일치하는 파일 없음

+2 -2
파일 보기
@@ -46,7 +46,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, true),
testing::Values("RCCL_ENABLE_CLIQUE=0", "RCCL_ENABLE_CLIQUE=1")),
@@ -71,7 +71,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, true),
testing::Values("RCCL_ENABLE_CLIQUE=0", "RCCL_ENABLE_CLIQUE=1")),