gtest: extend testing up to 8 GPUs

[ROCm/rccl commit: 8db0aa8f4c]

projects/rccl/test/test_AllGather.cpp

@@ -105,9 +105,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("")));

projects/rccl/test/test_AllReduce.cpp

@@ -55,7 +55,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, true),
                                 testing::Values("")));

projects/rccl/test/test_AllToAll.cpp

@@ -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")));

projects/rccl/test/test_Broadcast.cpp

@@ -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, true),
                                 testing::Values("")));

projects/rccl/test/test_CombinedCalls.cpp

@@ -91,9 +91,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("")));

projects/rccl/test/test_Gather.cpp

@@ -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")));

projects/rccl/test/test_GroupCalls.cpp

@@ -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("")));

projects/rccl/test/test_Reduce.cpp

@@ -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, true),
                                 testing::Values("")));

projects/rccl/test/test_ReduceScatter.cpp

@@ -59,9 +59,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("")));

projects/rccl/test/test_Scatter.cpp

@@ -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")));