diff --git a/test/test_AllGather.cpp b/test/test_AllGather.cpp index 558e927ffb..1e7b9441c0 100644 --- a/test/test_AllGather.cpp +++ b/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(""))); diff --git a/test/test_AllReduce.cpp b/test/test_AllReduce.cpp index 41f1af5a8b..1452ccdfef 100644 --- a/test/test_AllReduce.cpp +++ b/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(""))); diff --git a/test/test_AllToAll.cpp b/test/test_AllToAll.cpp index 2c1d29da1b..05d5e40475 100644 --- a/test/test_AllToAll.cpp +++ b/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"))); diff --git a/test/test_Broadcast.cpp b/test/test_Broadcast.cpp index 85b087a163..52e4403c75 100644 --- a/test/test_Broadcast.cpp +++ b/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(""))); diff --git a/test/test_CombinedCalls.cpp b/test/test_CombinedCalls.cpp index 35d61e82f3..1fe86a4125 100644 --- a/test/test_CombinedCalls.cpp +++ b/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(""))); diff --git a/test/test_Gather.cpp b/test/test_Gather.cpp index 4324434e9e..95dc35c461 100644 --- a/test/test_Gather.cpp +++ b/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"))); diff --git a/test/test_GroupCalls.cpp b/test/test_GroupCalls.cpp index cc2bd7dcd2..2f652e96e4 100644 --- a/test/test_GroupCalls.cpp +++ b/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(""))); diff --git a/test/test_Reduce.cpp b/test/test_Reduce.cpp index c181ec90ac..59dac111f9 100644 --- a/test/test_Reduce.cpp +++ b/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(""))); diff --git a/test/test_ReduceScatter.cpp b/test/test_ReduceScatter.cpp index 014ecd0ba4..ab4dc08860 100644 --- a/test/test_ReduceScatter.cpp +++ b/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(""))); diff --git a/test/test_Scatter.cpp b/test/test_Scatter.cpp index dca86b1ff3..80ac6e534c 100644 --- a/test/test_Scatter.cpp +++ b/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")));