Use aliases to check for installed UCX version (#692)

Fixes #691 Authors: - Peter Andreas Entschev (https://github.com/pentschev) Approvers: - Charles Blackmon-Luca (https://github.com/charlesbluca) - Benjamin Zaitlen (https://github.com/quasiben) URL: #692
rapidsai · Jul 28, 2021 · c9e3907 · c9e3907
1 parent a0217f4
commit c9e3907
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dask_cuda/tests/test_dgx.py b/dask_cuda/tests/test_dgx.py
@@ -367,7 +367,7 @@ def _timeout_callback():
 def test_dask_cuda_worker_ucx_net_devices(enable_rdmacm):
     ucp = pytest.importorskip("ucp")  # NOQA: F841
 
-    if ucp.get_ucx_version() >= (1, 10, 0):
+    if _ucx_110:
         pytest.skip("UCX 1.10 and higher should rely on default UCX_NET_DEVICES")
 
     p = mp.Process(

diff --git a/dask_cuda/utils.py b/dask_cuda/utils.py
@@ -192,7 +192,7 @@ def get_device_total_memory(index=0):
 def get_ucx_net_devices(
     cuda_device_index, ucx_net_devices, get_openfabrics=True, get_network=False
 ):
-    if ucp.get_ucx_version() >= (1, 11, 0) and ucx_net_devices == "auto":
+    if _ucx_111 and ucx_net_devices == "auto":
         return None
 
     if cuda_device_index is None and (