From 701c01ed5e98ceae29fda4ea2f14c7535ec2d5bc Mon Sep 17 00:00:00 2001 From: Olli Saarikivi Date: Wed, 18 Oct 2023 23:32:44 +0000 Subject: [PATCH] Revert testing change --- python/test/test_mscclpp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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")