gtest: extend testing up to 8 GPUs

[ROCm/rccl commit: 8db0aa8f4c]
Šī revīzija ir iekļauta:
Wenkai Du committed 2020-06-29 09:32:31 -07:00
1 vecāks 77d94493c3
revīzija ac87c9db37
10 files changed
+14 -14

No files matched your search

+2 -2
Parādīt failu
@@ -111,9 +111,9 @@ namespace CorrectnessTests
ncclFloat64,
ncclBfloat16),
// Number of elements
testing::Values(3072, 3145728),
testing::Values(2520, 3026520),
// Number of devices
testing::Values(2,3,4),
testing::Values(2,3,4,5,6,7,8),
// In-place or not
testing::Values(false, true),
testing::Values("")));