dask-cuda 25.8.0__py3-none-any.whl → 25.10.0__py3-none-any.whl

dask_cuda/tests/test_utils.py

@@ -24,7 +24,6 @@ from dask_cuda.utils import (
     parse_device_memory_limit,
     unpack_bitmask,
 )
-from dask_cuda.utils_test import get_ucx_implementation
 
 
 @patch.dict(os.environ, {"CUDA_VISIBLE_DEVICES": "0,1,2"})
@@ -88,15 +87,11 @@ def test_get_device_total_memory():
                 assert total_mem is None
 
 
-@pytest.mark.parametrize(
-    "protocol",
-    ["ucx", "ucx-old"],
-)
-def test_get_preload_options_default(protocol):
-    get_ucx_implementation(protocol)
+def test_get_preload_options_default():
+    pytest.importorskip("distributed_ucxx")
 
     opts = get_preload_options(
-        protocol=protocol,
+        protocol="ucx",
         create_cuda_context=True,
     )
 
@@ -106,18 +101,14 @@ def test_get_preload_options_default(protocol):
     assert opts["preload_argv"] == ["--create-cuda-context"]
 
 
-@pytest.mark.parametrize(
-    "protocol",
-    ["ucx", "ucx-old"],
-)
 @pytest.mark.parametrize("enable_tcp", [True, False])
 @pytest.mark.parametrize("enable_infiniband", [True, False])
 @pytest.mark.parametrize("enable_nvlink", [True, False])
-def test_get_preload_options(protocol, enable_tcp, enable_infiniband, enable_nvlink):
-    get_ucx_implementation(protocol)
+def test_get_preload_options(enable_tcp, enable_infiniband, enable_nvlink):
+    pytest.importorskip("distributed_ucxx")
 
     opts = get_preload_options(
-        protocol=protocol,
+        protocol="ucx",
         create_cuda_context=True,
         enable_tcp_over_ucx=enable_tcp,
         enable_infiniband=enable_infiniband,
@@ -137,17 +128,11 @@ def test_get_preload_options(protocol, enable_tcp, enable_infiniband, enable_nvl
         assert "--enable-nvlink" in opts["preload_argv"]
 
 
-@pytest.mark.parametrize(
-    "protocol",
-    ["ucx", "ucx-old"],
-)
 @pytest.mark.parametrize("enable_tcp_over_ucx", [True, False, None])
 @pytest.mark.parametrize("enable_nvlink", [True, False, None])
 @pytest.mark.parametrize("enable_infiniband", [True, False, None])
-def test_get_ucx_config(
-    protocol, enable_tcp_over_ucx, enable_infiniband, enable_nvlink
-):
-    get_ucx_implementation(protocol)
+def test_get_ucx_config(enable_tcp_over_ucx, enable_infiniband, enable_nvlink):
+    pytest.importorskip("distributed_ucxx")
 
     kwargs = {
         "enable_tcp_over_ucx": enable_tcp_over_ucx,

dask_cuda/tests/test_worker_spec.py

@@ -31,7 +31,7 @@ def _check_env_value(spec, k, v):
 @pytest.mark.parametrize("num_devices", [1, 4])
 @pytest.mark.parametrize("cls", [Nanny])
 @pytest.mark.parametrize("interface", [None, "eth0", "enp1s0f0"])
-@pytest.mark.parametrize("protocol", [None, "tcp", "ucx", "ucx-old"])
+@pytest.mark.parametrize("protocol", [None, "tcp", "ucx"])
 @pytest.mark.parametrize("dashboard_address", [None, ":0", ":8787"])
 @pytest.mark.parametrize("threads_per_worker", [1, 8])
 @pytest.mark.parametrize("silence_logs", [False, True])
@@ -48,6 +48,9 @@ def test_worker_spec(
     enable_infiniband,
     enable_nvlink,
 ):
+    if protocol == "ucx":
+        pytest.importorskip("distributed_ucxx")
+
     def _test():
         return worker_spec(
             CUDA_VISIBLE_DEVICES=list(range(num_devices)),
@@ -61,7 +64,7 @@ def test_worker_spec(
             enable_nvlink=enable_nvlink,
         )
 
-    if (enable_infiniband or enable_nvlink) and protocol not in ("ucx", "ucx-old"):
+    if (enable_infiniband or enable_nvlink) and protocol != "ucx":
         with pytest.raises(
             TypeError, match="Enabling InfiniBand or NVLink requires protocol='ucx'"
         ):

dask_cuda/utils.py

@@ -1,7 +1,6 @@
 # SPDX-FileCopyrightText: Copyright (c) 2019-2025, NVIDIA CORPORATION & AFFILIATES.
 # SPDX-License-Identifier: Apache-2.0
 
-import importlib
 import math
 import operator
 import os
@@ -19,7 +18,6 @@ import pynvml
 import toolz
 
 import dask
-import distributed  # noqa: required for dask.config.get("distributed.comm.ucx")
 from dask.config import canonical_name
 from dask.utils import format_bytes, parse_bytes
 from distributed import wait
@@ -266,17 +264,16 @@ def get_ucx_config(
     enable_infiniband=None,
     enable_nvlink=None,
     enable_rdmacm=None,
-    protocol=None,
 ):
-    ucx_config = dask.config.get("distributed.comm.ucx")
+    try:
+        import distributed_ucxx
+    except ImportError:
+        return None
 
-    # TODO: remove along with `protocol` kwarg when UCX-Py is removed, see
-    # https://github.com/rapidsai/dask-cuda/issues/1517
-    if protocol in ("ucx", "ucxx", "ucx-old"):
-        ucx_config[canonical_name("ucx-protocol", ucx_config)] = protocol
+    distributed_ucxx.config.setup_config()
+    ucx_config = dask.config.get("distributed-ucxx")
 
     ucx_config[canonical_name("create-cuda-context", ucx_config)] = True
-    ucx_config[canonical_name("reuse-endpoints", ucx_config)] = False
 
     # If any transport is explicitly disabled (`False`) by the user, others that
     # are not specified should be enabled (`True`). If transports are explicitly
@@ -358,11 +355,7 @@ def get_preload_options(
     if create_cuda_context:
         preload_options["preload_argv"].append("--create-cuda-context")
 
-    try:
-        _get_active_ucx_implementation_name(protocol)
-    except ValueError:
-        pass
-    else:
+    if protocol in ("ucx", "ucxx"):
         initialize_ucx_argv = []
         if enable_tcp_over_ucx:
             initialize_ucx_argv.append("--enable-tcp-over-ucx")
@@ -828,21 +821,12 @@ def get_worker_config(dask_worker):
             ret["device-memory-limit"] = dask_worker.data.device_buffer.n
 
     # using ucx ?
-    scheme, loc = parse_address(dask_worker.scheduler.address)
-    ret["protocol"] = scheme
-    try:
-        protocol = _get_active_ucx_implementation_name(scheme)
-    except ValueError:
-        pass
-    else:
-        if protocol == "ucxx":
-            import ucxx
-
-            ret["ucx-transports"] = ucxx.get_active_transports()
-        elif protocol == "ucx-old":
-            import ucp
+    protocol, loc = parse_address(dask_worker.scheduler.address)
+    ret["protocol"] = protocol
+    if protocol in ("ucx", "ucxx"):
+        import ucxx
 
-            ret["ucx-transports"] = ucp.get_active_transports()
+        ret["ucx-transports"] = ucxx.get_active_transports()
 
     # comm timeouts
     ret["distributed.comm.timeouts"] = dask.config.get("distributed.comm.timeouts")
@@ -988,41 +972,3 @@ class CommaSeparatedChoice(click.Choice):
                 choices_str = ", ".join(f"'{c}'" for c in self.choices)
                 self.fail(f"invalid choice(s): {v}. (choices are: {choices_str})")
         return values
-
-
-def _get_active_ucx_implementation_name(protocol):
-    """Get the name of active UCX implementation.
-
-    Determine what UCX implementation is being activated based on a series of
-    conditions. UCXX is selected if:
-    - The protocol is `"ucxx"`, or the protocol is `"ucx"` and the `distributed-ucxx`
-      package is installed.
-    UCX-Py is selected if:
-    - The protocol is `"ucx-old"`, or the protocol is `"ucx"` and the `distributed-ucxx`
-      package is not installed, in which case a `FutureWarning` is also raised.
-
-    Parameters
-    ----------
-    protocol: str
-        The communication protocol selected.
-
-    Returns
-    -------
-    The selected implementation type, either "ucxx" or "ucx-old".
-
-    Raises
-    ------
-    ValueError
-        If protocol is not a valid UCX protocol.
-    """
-    has_ucxx = importlib.util.find_spec("distributed_ucxx") is not None
-
-    if protocol == "ucxx" or (has_ucxx and protocol == "ucx"):
-        # With https://github.com/rapidsai/rapids-dask-dependency/pull/116,
-        # `protocol="ucx"` now points to UCXX (if distributed-ucxx is installed),
-        # thus call the UCXX initializer.
-        return "ucxx"
-    elif protocol in ("ucx", "ucx-old"):
-        return "ucx-old"
-    else:
-        raise ValueError("Protocol is neither UCXX nor UCX-Py")

dask_cuda/utils_test.py

@@ -6,8 +6,6 @@ from typing import Literal
 import distributed
 from distributed import Nanny, Worker
 
-from .utils import _get_active_ucx_implementation_name
-
 
 class MockWorker(Worker):
     """Mock Worker class preventing NVML from getting used by SystemMonitor.
@@ -48,14 +46,3 @@ class IncreasedCloseTimeoutNanny(Nanny):
         self, timeout: float = 30.0, reason: str = "nanny-close"
     ) -> Literal["OK"]:
         return await super().close(timeout=timeout, reason=reason)
-
-
-def get_ucx_implementation(protocol):
-    import pytest
-
-    protocol = _get_active_ucx_implementation_name(protocol)
-
-    if protocol == "ucxx":
-        return pytest.importorskip("ucxx")
-    else:
-        return pytest.importorskip("ucp")

dask_cuda/worker_spec.py

@@ -9,7 +9,7 @@ from distributed.system import MEMORY_LIMIT
 from .initialize import initialize
 from .local_cuda_cluster import cuda_visible_devices
 from .plugins import CPUAffinity
-from .utils import _get_active_ucx_implementation_name, get_cpu_affinity, get_gpu_count
+from .utils import get_cpu_affinity, get_gpu_count
 
 
 def worker_spec(
@@ -84,14 +84,12 @@ def worker_spec(
        'preload_argv': ['--create-cuda-context']}}}
 
     """
-    if enable_tcp_over_ucx or enable_infiniband or enable_nvlink:
-        try:
-            _get_active_ucx_implementation_name(protocol)
-        except ValueError:
-            raise TypeError(
-                "Enabling InfiniBand or NVLink requires protocol='ucx', "
-                "protocol='ucxx' or protocol='ucx-old'"
-            ) from None
+    if (
+        enable_tcp_over_ucx or enable_infiniband or enable_nvlink
+    ) and protocol != "ucx":
+        raise TypeError(
+            "Enabling InfiniBand or NVLink requires protocol='ucx'"
+        ) from None
 
     if CUDA_VISIBLE_DEVICES is None:
         CUDA_VISIBLE_DEVICES = os.environ.get(

{dask_cuda-25.8.0.dist-info → dask_cuda-25.10.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dask-cuda
-Version: 25.8.0
+Version: 25.10.0
 Summary: Utilities for Dask and CUDA interactions
 Author: NVIDIA Corporation
 License: Apache-2.0
@@ -20,17 +20,24 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: click>=8.1
-Requires-Dist: numba<0.62.0a0,>=0.59.1
+Requires-Dist: cuda-core==0.3.*
+Requires-Dist: numba-cuda<0.20.0a0,>=0.19.1
 Requires-Dist: numpy<3.0a0,>=1.23
+Requires-Dist: nvidia-ml-py>=12
 Requires-Dist: pandas>=1.3
-Requires-Dist: pynvml<13.0.0a0,>=12.0.0
-Requires-Dist: rapids-dask-dependency==25.8.*
+Requires-Dist: rapids-dask-dependency==25.10.*
 Requires-Dist: zict>=2.0.0
 Provides-Extra: docs
 Requires-Dist: numpydoc>=1.1.0; extra == "docs"
 Requires-Dist: sphinx; extra == "docs"
 Requires-Dist: sphinx-click>=2.7.1; extra == "docs"
 Requires-Dist: sphinx-rtd-theme>=0.5.1; extra == "docs"
+Provides-Extra: cu12
+Requires-Dist: cuda-core[cu12]==0.3.*; extra == "cu12"
+Requires-Dist: numba-cuda[cu12]<0.20.0a0,>=0.19.1; extra == "cu12"
+Provides-Extra: cu13
+Requires-Dist: cuda-core[cu13]==0.3.*; extra == "cu13"
+Requires-Dist: numba-cuda[cu13]<0.20.0a0,>=0.19.1; extra == "cu13"
 Dynamic: license-file
 
 Dask CUDA

dask_cuda-25.10.0.dist-info/RECORD

@@ -0,0 +1,63 @@
+dask_cuda/GIT_COMMIT,sha256=b9kAPQayKE7_CmJ29dDIOd-qHhFKihEapaSOkl84OFU,41
+dask_cuda/VERSION,sha256=qFUmAhDerTzlTeG7G1N8kkXh-gyZzDU9R0ARLQtoP_Y,8
+dask_cuda/__init__.py,sha256=Wbc7R0voN4vsQkb7SKuVXH0YXuXtfnAxrupxfM4lT10,1933
+dask_cuda/_compat.py,sha256=AG2lKGAtZitDPBjHeFDKLTN_B5HKodrhZ2kHlk1Z-D0,498
+dask_cuda/_version.py,sha256=cHDO9AzNtxkCVhwYu7hL3H7RPAkQnxpKBjElOst3rkI,964
+dask_cuda/cli.py,sha256=VRYuryhViVWkCH7H7fDDTMToSOC17nAUMIPbd3K2jRs,17490
+dask_cuda/cuda_worker.py,sha256=OwAi8Qv45a-KTBC2XJdRdd-iuId3vb0bTTONOuq9DEQ,8586
+dask_cuda/device_host_file.py,sha256=wTiTyYthUrR8l2WM7iV5lvjQDzeYxnnK7GUlPsHI6p4,11042
+dask_cuda/disk_io.py,sha256=IpD2hA-AjTw3SEu3w9pT9ELAUpKW0XcphPh8WwlKH70,6757
+dask_cuda/get_device_memory_objects.py,sha256=je10ygTAiifDxZupmyr_wzkbDuf9n3_saQGdDJAUwig,4483
+dask_cuda/initialize.py,sha256=DH8_zOmK81_kygijCvAVdvcBz9QvH7IwhLueNaRGuos,8035
+dask_cuda/is_device_object.py,sha256=x9klFdeQzLcug7wZMxN3GK2AS121tlDe-LQ2uznm5yo,1179
+dask_cuda/is_spillable_object.py,sha256=8gj6QgtKcmzrpQwy8rE-pS1R8tjaJOeD-Fzr6LumjJg,1596
+dask_cuda/local_cuda_cluster.py,sha256=ihf0jTUAyRuKJZPgmvvZ0ZLsgp-No9QuxGZzRBdQWNU,20024
+dask_cuda/plugins.py,sha256=u4gWQy8DgaTAsd59KCNXLG6zmdmGg5qhaI8ha2rMEFs,7085
+dask_cuda/proxify_device_objects.py,sha256=hGHK2gBkuaMYyrZqcitRGUiH_up56R1hsC7b_6YcCBU,8325
+dask_cuda/proxify_host_file.py,sha256=TbdszPvAgV4CRPMz99tumnCWwiCTqMBZqHQua3RRpHE,31031
+dask_cuda/proxy_object.py,sha256=eII-S0vdpFa6NLebpHFVQuJ2eyXJ5cBg8sucgFEa91g,30307
+dask_cuda/utils.py,sha256=981BxG2hOcjgr5FNMIdOsKTC_jyTi8D7HC3lHWkC8wM,31518
+dask_cuda/utils_test.py,sha256=U-_CRo7bkRJh5bDrCY9pGNKnU1PlJ4VA1gzrbh8Ubv4,1833
+dask_cuda/worker_common.py,sha256=uXoYZ1IZanAbHddfpL3NbVDV5WqCF0m94nOiqqvEnxc,7168
+dask_cuda/worker_spec.py,sha256=JNjY46iIR1OkanCfKBEcusVFgv33u2Lc2Pg7Lb30HfU,4531
+dask_cuda/benchmarks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+dask_cuda/benchmarks/common.py,sha256=4AHgWkomo1RsPwH6eogz4vbE9vg_Dg-krScn9B1BRuw,7057
+dask_cuda/benchmarks/local_cudf_groupby.py,sha256=9jmSnP0A6Q8MR7j5nCGtNivw9S1jh1XD7lRQge9AtRQ,8635
+dask_cuda/benchmarks/local_cudf_merge.py,sha256=XCBaPxydH2DGzetp7cN3u5W3bnkOg_wfo3RQYz0GHDY,12330
+dask_cuda/benchmarks/local_cudf_shuffle.py,sha256=kqtA6eoC1r5TTEEDm48zuS8BoK05NYDSInpkfFcwvYQ,8375
+dask_cuda/benchmarks/local_cupy.py,sha256=jqq-wGsVmvZuKmtRu7smvRmnXL_jCksj083zCOKEBOA,10446
+dask_cuda/benchmarks/local_cupy_map_overlap.py,sha256=-6UoiH5PZgRRRClj6NoUTLCd4nvhAZMtj9wWK2tsqkA,6140
+dask_cuda/benchmarks/read_parquet.py,sha256=spKu6RLWYngPZq9hnaoU0mz7INIaJnErfqjBG2wH8Zc,7614
+dask_cuda/benchmarks/utils.py,sha256=o7RoUZnbHgib5KcR7-m86x3F6tIUGbgq_a45PMHURwg,30224
+dask_cuda/explicit_comms/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+dask_cuda/explicit_comms/comms.py,sha256=FbSDPQUz0gwQNa8EQMNEGEwWCM2KhuDk4CiNNjZy6qo,11611
+dask_cuda/explicit_comms/dataframe/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+dask_cuda/explicit_comms/dataframe/shuffle.py,sha256=vq-NBPYtKvKcfo7-eikPYaeWDf-V_DTHR5PVE_E-dyE,23863
+dask_cuda/tests/conftest.py,sha256=NjTnsgu7UewTpoTakt-2bpe0rtarwni25anjZPWbbb0,1501
+dask_cuda/tests/test_cudf_builtin_spilling.py,sha256=fhkLAyLBbDmS5pbyR1XbafS_OVW0Z6oVc3GerfPY0wg,5029
+dask_cuda/tests/test_dask_cuda_worker.py,sha256=iOkYAGSYfql7Wxxnb1hZJ7oedQyLmweXAHpQhckaTgY,23232
+dask_cuda/tests/test_dask_setup.py,sha256=gwu_RXIw3qMQExzUjIWzsClMavJEAQwZj9SgZCbF960,6741
+dask_cuda/tests/test_device_host_file.py,sha256=79ssUISo1YhsW_7HdwqPfsH2LRzS2bi5BjPym1Sdgqw,5882
+dask_cuda/tests/test_dgx.py,sha256=ZyzLelIbzELv_-gnh_CP6uedAVqfb5yzaXuT0GD-7WM,6700
+dask_cuda/tests/test_explicit_comms.py,sha256=FzK0wOaR3y6jMp-7Lj0xrhIhRRlpfGBEstBGqYbBMrg,20551
+dask_cuda/tests/test_from_array.py,sha256=AtGWl0Q9-u53hEGgGcwO65nZ0lCvybRIFHZP2p499EU,589
+dask_cuda/tests/test_gds.py,sha256=j1Huud6UGm1fbkyRLQEz_ysrVw__5AimwSn_M-2GEvs,1513
+dask_cuda/tests/test_initialize.py,sha256=fVlA8I1XnPMg3h5ARDKiVC8zzmYpTBNtDgiBgnazBwA,14751
+dask_cuda/tests/test_local_cuda_cluster.py,sha256=LDet1NCYCX2wNEhun1B9KflJhajUjwHgzE_QE-RY_1A,21819
+dask_cuda/tests/test_proxify_host_file.py,sha256=EdU75aZ_805ggK00wC46DMW5iJ7lcGtm9xgPoEHU_Jc,20123
+dask_cuda/tests/test_proxy.py,sha256=HyjwevaeJIsyqg9RpbzJsVp9VquDQfupHcanXSECzcs,23933
+dask_cuda/tests/test_spill.py,sha256=IzOOqb5hDYM2OntZZBavy1fwB6ShWmbmZw7zHE0w6Ug,15760
+dask_cuda/tests/test_utils.py,sha256=wrCRRN9zpk1NarYPLyB0da6axSSzfjsdFp6gufVFjO4,12018
+dask_cuda/tests/test_version.py,sha256=vK2HjlRLX0nxwvRsYxBqhoZryBNZklzA-vdnyuWDxVg,365
+dask_cuda/tests/test_worker_spec.py,sha256=gO0lLJqoPH_Ti6YYrj5k56h_ux-QW2Zrc_MuKPei-K4,2597
+dask_cuda-25.10.0.dist-info/licenses/LICENSE,sha256=MjI3I-EgxfEvZlgjk82rgiFsZqSDXHFETd2QJ89UwDA,11348
+examples/ucx/client_initialize.py,sha256=YN3AXHF8btcMd6NicKKhKR9SXouAsK1foJhFspbOn70,1262
+examples/ucx/local_cuda_cluster.py,sha256=7xVY3EhwhkY2L4VZin_BiMCbrjhirDNChoC86KiETNc,1983
+shared-actions/check_nightly_success/check-nightly-success/check.py,sha256=AK7qCq6OcXnoOhJSOUc_aIl4VA1uo6nK5nJtFoZwNJQ,5371
+shared-actions/telemetry-impls/summarize/bump_time.py,sha256=FXcGDqjCabsL6vlqd5RIV2PCWi2ns_ju8Ul0ERM21cA,2033
+shared-actions/telemetry-impls/summarize/send_trace.py,sha256=i30O_cKZ1OtXIbqaTE3R1JLJSw2XoYdmoVpM2Nc9wj8,16555
+dask_cuda-25.10.0.dist-info/METADATA,sha256=fYduQsSFn4GsVscPIj2spybnzVf0HXmh0cHF59E6LdE,2662
+dask_cuda-25.10.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dask_cuda-25.10.0.dist-info/entry_points.txt,sha256=UcRaKVEpywtxc6pF1VnfMB0UK4sJg7a8_NdZF67laPM,136
+dask_cuda-25.10.0.dist-info/top_level.txt,sha256=SaQxjNeXM9ZhEBjU6CY25OIOt2UojSREGiVxg9UIc08,63
+dask_cuda-25.10.0.dist-info/RECORD,,

shared-actions/check_nightly_success/check-nightly-success/check.py

@@ -72,7 +72,7 @@ def main(
 
     for branch, branch_runs in branch_dict.items():
         # Only consider RAPIDS release branches, which have versions like
-        # '25.02' (RAPIDS) or '0.42' (ucxx, ucx-py).
+        # '25.10' (RAPIDS) or '0.46' (ucxx).
         if not re.match("branch-[0-9]{1,2}.[0-9]{2}", branch):
             continue
 

dask_cuda/tests/test_rdd_ucx.py

@@ -1,160 +0,0 @@
-# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES.
-# SPDX-License-Identifier: Apache-2.0
-
-
-import importlib
-import io
-import multiprocessing as mp
-import sys
-
-import pytest
-
-from dask_cuda import LocalCUDACluster
-
-mp = mp.get_context("spawn")  # type: ignore
-
-
-def _has_distributed_ucxx() -> bool:
-    return bool(importlib.util.find_spec("distributed_ucxx"))
-
-
-def _test_protocol_ucx():
-    with LocalCUDACluster(protocol="ucx") as cluster:
-        assert cluster.scheduler_comm.address.startswith("ucx://")
-
-        if _has_distributed_ucxx():
-            import distributed_ucxx
-
-            assert all(
-                isinstance(batched_send.comm, distributed_ucxx.ucxx.UCXX)
-                for batched_send in cluster.scheduler.stream_comms.values()
-            )
-        else:
-            import rapids_dask_dependency
-
-            assert all(
-                isinstance(
-                    batched_send.comm,
-                    rapids_dask_dependency.patches.distributed.comm.__rdd_patch_ucx.UCX,
-                )
-                for batched_send in cluster.scheduler.stream_comms.values()
-            )
-
-
-def _test_protocol_ucxx():
-    if _has_distributed_ucxx():
-        with LocalCUDACluster(protocol="ucxx") as cluster:
-            assert cluster.scheduler_comm.address.startswith("ucxx://")
-            import distributed_ucxx
-
-            assert all(
-                isinstance(batched_send.comm, distributed_ucxx.ucxx.UCXX)
-                for batched_send in cluster.scheduler.stream_comms.values()
-            )
-    else:
-        with pytest.raises(RuntimeError, match="Cluster failed to start"):
-            LocalCUDACluster(protocol="ucxx")
-
-
-def _test_protocol_ucx_old():
-    with LocalCUDACluster(protocol="ucx-old") as cluster:
-        assert cluster.scheduler_comm.address.startswith("ucx-old://")
-
-        import rapids_dask_dependency
-
-        assert all(
-            isinstance(
-                batched_send.comm,
-                rapids_dask_dependency.patches.distributed.comm.__rdd_patch_ucx.UCX,
-            )
-            for batched_send in cluster.scheduler.stream_comms.values()
-        )
-
-
-def _run_test_with_output_capture(test_func_name, conn):
-    """Run a test function in a subprocess and capture stdout/stderr."""
-    # Redirect stdout and stderr to capture output
-    old_stdout = sys.stdout
-    old_stderr = sys.stderr
-    captured_output = io.StringIO()
-    sys.stdout = sys.stderr = captured_output
-
-    try:
-        # Import and run the test function
-        if test_func_name == "_test_protocol_ucx":
-            _test_protocol_ucx()
-        elif test_func_name == "_test_protocol_ucxx":
-            _test_protocol_ucxx()
-        elif test_func_name == "_test_protocol_ucx_old":
-            _test_protocol_ucx_old()
-        else:
-            raise ValueError(f"Unknown test function: {test_func_name}")
-
-        output = captured_output.getvalue()
-        conn.send((True, output))  # True = success
-    except Exception as e:
-        output = captured_output.getvalue()
-        output += f"\nException: {e}"
-        import traceback
-
-        output += f"\nTraceback:\n{traceback.format_exc()}"
-        conn.send((False, output))  # False = failure
-    finally:
-        # Restore original stdout/stderr
-        sys.stdout = old_stdout
-        sys.stderr = old_stderr
-        conn.close()
-
-
-@pytest.mark.parametrize("protocol", ["ucx", "ucxx", "ucx-old"])
-def test_rdd_protocol(protocol):
-    """Test rapids-dask-dependency protocol selection"""
-    if protocol == "ucx":
-        test_func_name = "_test_protocol_ucx"
-    elif protocol == "ucxx":
-        test_func_name = "_test_protocol_ucxx"
-    else:
-        test_func_name = "_test_protocol_ucx_old"
-
-    # Create a pipe for communication between parent and child processes
-    parent_conn, child_conn = mp.Pipe()
-    p = mp.Process(
-        target=_run_test_with_output_capture, args=(test_func_name, child_conn)
-    )
-
-    p.start()
-    p.join(timeout=60)
-
-    if p.is_alive():
-        p.kill()
-        p.close()
-        raise TimeoutError("Test process timed out")
-
-    # Get the result from the child process
-    success, output = parent_conn.recv()
-
-    # Check that the test passed
-    assert success, f"Test failed in subprocess. Output:\n{output}"
-
-    # For the ucx protocol, check if warnings are printed when distributed_ucxx is not
-    # available
-    if protocol == "ucx" and not _has_distributed_ucxx():
-        # Check if the warning about protocol='ucx' is printed
-        print(f"Output for {protocol} protocol:\n{output}")
-        assert (
-            "you have requested protocol='ucx'" in output
-        ), f"Expected warning not found in output: {output}"
-        assert (
-            "'distributed-ucxx' is not installed" in output
-        ), f"Expected warning about distributed-ucxx not found in output: {output}"
-    elif protocol == "ucx" and _has_distributed_ucxx():
-        # When distributed_ucxx is available, the warning should NOT be printed
-        assert "you have requested protocol='ucx'" not in output, (
-            "Warning should not be printed when distributed_ucxx is available: "
-            f"{output}"
-        )
-    elif protocol == "ucx-old":
-        # The ucx-old protocol should not generate warnings
-        assert (
-            "you have requested protocol='ucx'" not in output
-        ), f"Warning should not be printed for ucx-old protocol: {output}"

dask_cuda-25.8.0.dist-info/RECORD

@@ -1,63 +0,0 @@
-dask_cuda/GIT_COMMIT,sha256=RAxzl5uQmWcKe5o7s9iHwueNfKc91R_-kniXtpRqmng,41
-dask_cuda/VERSION,sha256=mZz9G1Ul4kEOksaMu07UE-AVtGzT1t91nQu3CK9KUAk,8
-dask_cuda/__init__.py,sha256=Wbc7R0voN4vsQkb7SKuVXH0YXuXtfnAxrupxfM4lT10,1933
-dask_cuda/_compat.py,sha256=AG2lKGAtZitDPBjHeFDKLTN_B5HKodrhZ2kHlk1Z-D0,498
-dask_cuda/_version.py,sha256=cHDO9AzNtxkCVhwYu7hL3H7RPAkQnxpKBjElOst3rkI,964
-dask_cuda/cli.py,sha256=VRYuryhViVWkCH7H7fDDTMToSOC17nAUMIPbd3K2jRs,17490
-dask_cuda/cuda_worker.py,sha256=7ZLZ3AY3l1fLumx2XynUOej5Sx6bwZQRomT4Nj9pbyA,8590
-dask_cuda/device_host_file.py,sha256=wTiTyYthUrR8l2WM7iV5lvjQDzeYxnnK7GUlPsHI6p4,11042
-dask_cuda/disk_io.py,sha256=IpD2hA-AjTw3SEu3w9pT9ELAUpKW0XcphPh8WwlKH70,6757
-dask_cuda/get_device_memory_objects.py,sha256=Nk0f5kv2f0e6JSd5hrwenAOARgF4NS21Zv_x3zF3ONI,4577
-dask_cuda/initialize.py,sha256=JBgXNzt52OzcusC89k0eVn2ivbCoCzZxOPNvQgj3YfE,7385
-dask_cuda/is_device_object.py,sha256=x9klFdeQzLcug7wZMxN3GK2AS121tlDe-LQ2uznm5yo,1179
-dask_cuda/is_spillable_object.py,sha256=8gj6QgtKcmzrpQwy8rE-pS1R8tjaJOeD-Fzr6LumjJg,1596
-dask_cuda/local_cuda_cluster.py,sha256=pocJgHbs8h2z_hfChU2_s7kwYKuYTgFZtmrEgYHjWwc,20735
-dask_cuda/plugins.py,sha256=u4gWQy8DgaTAsd59KCNXLG6zmdmGg5qhaI8ha2rMEFs,7085
-dask_cuda/proxify_device_objects.py,sha256=hGHK2gBkuaMYyrZqcitRGUiH_up56R1hsC7b_6YcCBU,8325
-dask_cuda/proxify_host_file.py,sha256=TbdszPvAgV4CRPMz99tumnCWwiCTqMBZqHQua3RRpHE,31031
-dask_cuda/proxy_object.py,sha256=eII-S0vdpFa6NLebpHFVQuJ2eyXJ5cBg8sucgFEa91g,30307
-dask_cuda/utils.py,sha256=smUv6DGJlHgXjfASVJMlRL1NgECiwwQ2zgn_FCWxMQ4,33454
-dask_cuda/utils_test.py,sha256=CKRMB5KUAg-7VSf21AU1TQoqGpJXm9ftgRV7mGIGQ3s,2132
-dask_cuda/worker_common.py,sha256=uXoYZ1IZanAbHddfpL3NbVDV5WqCF0m94nOiqqvEnxc,7168
-dask_cuda/worker_spec.py,sha256=cI4vS08gyrIU3PKJIjjWZNXChUm5Pv9LwaShPqYYMUQ,4698
-dask_cuda/benchmarks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dask_cuda/benchmarks/common.py,sha256=4AHgWkomo1RsPwH6eogz4vbE9vg_Dg-krScn9B1BRuw,7057
-dask_cuda/benchmarks/local_cudf_groupby.py,sha256=ehKOJqnHgQoEyNWuebIWdJP8f_bRiwBd4pax3hkQP_8,8646
-dask_cuda/benchmarks/local_cudf_merge.py,sha256=D2yXcEj1dLvfdd-ugV6kLgcUoiHzLmN3jNNs_BjaWME,12341
-dask_cuda/benchmarks/local_cudf_shuffle.py,sha256=MxUHzif_7TFPj96yE-Y3-U_NMXD6VJLNB1fB39LXgrk,8386
-dask_cuda/benchmarks/local_cupy.py,sha256=1riE9_hVkDxfSP2pWU0h2VF9HTXhF3C2ArKb19KMnVo,10457
-dask_cuda/benchmarks/local_cupy_map_overlap.py,sha256=27OtGOhmHIQhHSAmyOm0GgUX8u10npmwBo_q5fLAqUM,6151
-dask_cuda/benchmarks/read_parquet.py,sha256=spKu6RLWYngPZq9hnaoU0mz7INIaJnErfqjBG2wH8Zc,7614
-dask_cuda/benchmarks/utils.py,sha256=pl0VG8BQnIrVvpdVQtP8ybX1SOcv1E01jZzhkfOIPw4,30235
-dask_cuda/explicit_comms/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dask_cuda/explicit_comms/comms.py,sha256=FbSDPQUz0gwQNa8EQMNEGEwWCM2KhuDk4CiNNjZy6qo,11611
-dask_cuda/explicit_comms/dataframe/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dask_cuda/explicit_comms/dataframe/shuffle.py,sha256=vq-NBPYtKvKcfo7-eikPYaeWDf-V_DTHR5PVE_E-dyE,23863
-dask_cuda/tests/conftest.py,sha256=NjTnsgu7UewTpoTakt-2bpe0rtarwni25anjZPWbbb0,1501
-dask_cuda/tests/test_cudf_builtin_spilling.py,sha256=qVN9J0Hdv66A9COFArLIdRriyyxEKpS3lEZGHbVHaq8,4903
-dask_cuda/tests/test_dask_cuda_worker.py,sha256=iOkYAGSYfql7Wxxnb1hZJ7oedQyLmweXAHpQhckaTgY,23232
-dask_cuda/tests/test_device_host_file.py,sha256=79ssUISo1YhsW_7HdwqPfsH2LRzS2bi5BjPym1Sdgqw,5882
-dask_cuda/tests/test_dgx.py,sha256=Y7OgFlrq_BzoF4L8qoA5byKX1kDA1foaFzT-gEfhrsI,7294
-dask_cuda/tests/test_explicit_comms.py,sha256=ULrVhOcVavEWyZBxVSXDOFT_nznF8mRrU1M3fq_DUjo,20622
-dask_cuda/tests/test_from_array.py,sha256=-mXVwrw3MjB32vFfdUvxiwojqu9mEYNyjZjGPg1h1sE,733
-dask_cuda/tests/test_gds.py,sha256=j1Huud6UGm1fbkyRLQEz_ysrVw__5AimwSn_M-2GEvs,1513
-dask_cuda/tests/test_initialize.py,sha256=2Fm84sAN0A9Kxhm7PiWgrwppNCaJLA2yj65Y3LPOrNw,7542
-dask_cuda/tests/test_local_cuda_cluster.py,sha256=wYhY4OzQcYDtevL2peT6k5bAmZs2MDpQMrwPs2To7AY,23083
-dask_cuda/tests/test_proxify_host_file.py,sha256=6YPQia-V5tcVOy2ZNwL9ZpD6qKxKCcaNZZK-8f4aqpM,19996
-dask_cuda/tests/test_proxy.py,sha256=3uREA-d4y0lmGHKGJ5z9OiAYTBie1Y3bpB8djQRrluE,23996
-dask_cuda/tests/test_rdd_ucx.py,sha256=nMbEm7kxAATHfVyZOkfpexyk-ZKtCbLzvPHdspK877M,5433
-dask_cuda/tests/test_spill.py,sha256=QmBdFBvmZsDbOo0IAeUAJQfMyx-BBxokosch4gZHa00,15704
-dask_cuda/tests/test_utils.py,sha256=bTpiNQyKTxLHzHTLn-G0SWVgetq_tqUGq7rbafvdwgg,12297
-dask_cuda/tests/test_version.py,sha256=vK2HjlRLX0nxwvRsYxBqhoZryBNZklzA-vdnyuWDxVg,365
-dask_cuda/tests/test_worker_spec.py,sha256=dlZ4OIOl7CVm4euMtvCUif1QetCWcntSRdwzAmkH6ec,2550
-dask_cuda-25.8.0.dist-info/licenses/LICENSE,sha256=MjI3I-EgxfEvZlgjk82rgiFsZqSDXHFETd2QJ89UwDA,11348
-examples/ucx/client_initialize.py,sha256=YN3AXHF8btcMd6NicKKhKR9SXouAsK1foJhFspbOn70,1262
-examples/ucx/local_cuda_cluster.py,sha256=7xVY3EhwhkY2L4VZin_BiMCbrjhirDNChoC86KiETNc,1983
-shared-actions/check_nightly_success/check-nightly-success/check.py,sha256=lBhwgJALfUXdk4B9IhYf1AV0OUu0dQol3GtUe-CnUgY,5379
-shared-actions/telemetry-impls/summarize/bump_time.py,sha256=FXcGDqjCabsL6vlqd5RIV2PCWi2ns_ju8Ul0ERM21cA,2033
-shared-actions/telemetry-impls/summarize/send_trace.py,sha256=i30O_cKZ1OtXIbqaTE3R1JLJSw2XoYdmoVpM2Nc9wj8,16555
-dask_cuda-25.8.0.dist-info/METADATA,sha256=euY_vTYPba5yD4W8FtKDhCAjF4qs0ZU6-z5cs2lxYQg,2345
-dask_cuda-25.8.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dask_cuda-25.8.0.dist-info/entry_points.txt,sha256=UcRaKVEpywtxc6pF1VnfMB0UK4sJg7a8_NdZF67laPM,136
-dask_cuda-25.8.0.dist-info/top_level.txt,sha256=SaQxjNeXM9ZhEBjU6CY25OIOt2UojSREGiVxg9UIc08,63
-dask_cuda-25.8.0.dist-info/RECORD,,