gtest: extend testing up to 8 GPUs

This commit is contained in:
Wenkai Du
2020-06-27 21:30:50 +00:00
parent 964c4c2061
commit 8db0aa8f4c
10 changed files with 14 additions and 14 deletions
+1 -1
View File
@@ -63,7 +63,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")));