Skip DGX InfiniBand tests when "rc" transport unavailable

rapidsai · Aug 9, 2021 · 0a95914 · 0a95914
1 parent b3bda5d
commit 0a95914
Showing 1 changed file with 7 additions and 0 deletions.
diff --git 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,),
     )