gtest: extend testing up to 8 GPUs

This commit is contained in:
Wenkai Du committed 2020-06-29 09:32:31 -07:00
1 parent 964c4c2061
commit 8db0aa8f4c
10 files changed
+14 -14

No files matched your search

+1 -1
View File
@@ -59,7 +59,7 @@ namespace CorrectnessTests
// Number of elements
testing::Values(1024, 1048576),
// Number of devices
testing::Values(2,3,4),
testing::Values(2,3,4,5,6,7,8),
// In-place or not
testing::Values(false),
testing::Values("RCCL_ALLTOALL_KERNEL_DISABLE=0", "RCCL_ALLTOALL_KERNEL_DISABLE=1")));