From 0a959148743334d4ef2343464c3103d4cf726d43 Mon Sep 17 00:00:00 2001 From: Peter Andreas Entschev Date: Mon, 9 Aug 2021 15:06:58 -0700 Subject: [PATCH] Skip DGX InfiniBand tests when "rc" transport unavailable --- dask_cuda/tests/test_dgx.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/dask_cuda/tests/test_dgx.py b/dask_cuda/tests/test_dgx.py index 7c029b7e7..9ec67b4b9 100644 --- a/dask_cuda/tests/test_dgx.py +++ b/dask_cuda/tests/test_dgx.py @@ -249,6 +249,10 @@ def check_ucx_options(): def test_ucx_infiniband_nvlink(params): ucp = pytest.importorskip("ucp") # NOQA: F841 + if params["enable_infiniband"]: + if not any([at.startswith("rc") for at in ucp.get_active_transports()]): + pytest.skip("No support available for 'rc' transport in UCX") + p = mp.Process( target=_test_ucx_infiniband_nvlink, args=( @@ -370,6 +374,9 @@ def test_dask_cuda_worker_ucx_net_devices(enable_rdmacm): if _ucx_110: pytest.skip("UCX 1.10 and higher should rely on default UCX_NET_DEVICES") + if not any([at.startswith("rc") for at in ucp.get_active_transports()]): + pytest.skip("No support available for 'rc' transport in UCX") + p = mp.Process( target=_test_dask_cuda_worker_ucx_net_devices, args=(enable_rdmacm,), )