Increase maximum UCX runtime pin

conda/recipes/ucx-py/meta.yaml

@@ -39,7 +39,7 @@ requirements:
     {% endfor %}
   run:
     - python
-    - ucx >=1.15.0,<1.16.0
+    - ucx >=1.15.0,<1.18.0
     # 'libucx' wheel dependency is unnecessary... the 'ucx' conda-forge package is used here instead
     {% for req in data["project"]["dependencies"] if not req.startswith("libucx") %}
     - {{ req }}

dependencies.yaml

@@ -157,7 +157,7 @@ dependencies:
     common:
       - output_types: conda
         packages:
-          - ucx>=1.15.0,<1.16
+          - ucx>=1.15.0,<1.18
       - output_types: requirements
         packages:
           # pip recognizes the index as a global option for the requirements.txt file
@@ -168,13 +168,13 @@ dependencies:
         matrices:
           - matrix: {cuda: "12.*"}
             packages:
-              - libucx-cu12>=1.15.0,<1.16
+              - libucx-cu12>=1.15.0,<1.18
           - matrix: {cuda: "11.*"}
             packages:
-              - libucx-cu11>=1.15.0,<1.16
+              - libucx-cu11>=1.15.0,<1.18
           - matrix: null
             packages:
-              - libucx>=1.15.0,<1.16
+              - libucx>=1.15.0,<1.18
   test_python:
     common:
       - output_types: [conda, requirements, pyproject]

pyproject.toml

@@ -30,7 +30,7 @@ authors = [
 license = { text = "BSD-3-Clause" }
 requires-python = ">=3.9"
 dependencies = [
-    "libucx>=1.15.0,<1.16",
+    "libucx>=1.15.0,<1.18",
     "numpy>=1.23,<2.0a0",
     "pynvml>=11.4.1",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`.

ucp/__init__.py

@@ -102,6 +102,11 @@ def _is_mig_device(handle):
     logger.info("Setting UCX_MAX_RNDV_RAILS=1")
     os.environ["UCX_MAX_RNDV_RAILS"] = "1"
 
+if "UCX_PROTO_ENABLE" not in os.environ:
+    # UCX protov2 still doesn't support CUDA async/managed memory
+    logger.info("Setting UCX_PROTO_ENABLE=n")
+    os.environ["UCX_PROTO_ENABLE"] = "n"
+
 
 __ucx_version__ = "%d.%d.%d" % get_ucx_version()