Revert testing change

This commit is contained in:
Olli Saarikivi
2023-10-18 23:32:44 +00:00
committed by Saeed Maleki
parent 48d9384959
commit 701c01ed5e

View File

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