diff --git a/tests/v1/kv_connector/unit/test_moriio_connector.py b/tests/v1/kv_connector/unit/test_moriio_connector.py index 1eca4964fd6c..76ca9e14a95d 100644 --- a/tests/v1/kv_connector/unit/test_moriio_connector.py +++ b/tests/v1/kv_connector/unit/test_moriio_connector.py @@ -46,6 +46,21 @@ ) +def is_ibverbs_available(): + from shutil import which + + if which("ibv_devices") is None: + return False + from subprocess import check_output + + output = check_output(["ibv_devices"]).decode("UTF-8") + return len(output.split("\n")) > 3 + + +if not is_ibverbs_available(): + pytest.skip("This test requires IB Verbs to run", allow_module_level=True) + + @pytest.fixture def mock_parallel_groups(): """Mock tensor/data parallel group functions for single-rank tests."""