diff --git a/python/test/test_mscclpp.py b/python/test/test_mscclpp.py index e5538553..70c7194c 100644 --- a/python/test/test_mscclpp.py +++ b/python/test/test_mscclpp.py @@ -309,7 +309,7 @@ def test_d2d_semaphores(mpi_group: MpiGroup): @parametrize_mpi_groups(2, 4, 8, 16) -@pytest.mark.parametrize("nelem", [2**i for i in [10]]) +@pytest.mark.parametrize("nelem", [2**i for i in [10, 15, 20]]) @pytest.mark.parametrize("use_packet", [False, True]) def test_sm_channels(mpi_group: MpiGroup, nelem: int, use_packet: bool): group, connections = create_and_connect(mpi_group, "NVLink")