Update to numba-cuda >=0.22.1,<0.23.0

conda/environments/all_cuda-129_arch-aarch64.yaml

@@ -55,7 +55,7 @@ dependencies:
 - nbsphinx
 - ninja
 - notebook
-- numba-cuda>=0.19.1,<0.20.0a0
+- numba-cuda>=0.22.1,<0.23.0
 - numba>=0.60.0,<0.62.0a0
 - numpy>=1.23,<3.0a0
 - numpydoc

conda/environments/all_cuda-129_arch-x86_64.yaml

@@ -56,7 +56,7 @@ dependencies:
 - nbsphinx
 - ninja
 - notebook
-- numba-cuda>=0.19.1,<0.20.0a0
+- numba-cuda>=0.22.1,<0.23.0
 - numba>=0.60.0,<0.62.0a0
 - numpy>=1.23,<3.0a0
 - numpydoc

conda/environments/all_cuda-130_arch-aarch64.yaml

@@ -55,7 +55,7 @@ dependencies:
 - nbsphinx
 - ninja
 - notebook
-- numba-cuda>=0.19.1,<0.20.0a0
+- numba-cuda>=0.22.1,<0.23.0
 - numba>=0.60.0,<0.62.0a0
 - numpy>=1.23,<3.0a0
 - numpydoc

conda/environments/all_cuda-130_arch-x86_64.yaml

@@ -56,7 +56,7 @@ dependencies:
 - nbsphinx
 - ninja
 - notebook
-- numba-cuda>=0.19.1,<0.20.0a0
+- numba-cuda>=0.22.1,<0.23.0
 - numba>=0.60.0,<0.62.0a0
 - numpy>=1.23,<3.0a0
 - numpydoc

conda/recipes/cudf/recipe.yaml

@@ -71,7 +71,7 @@ requirements:
     - rapids-build-backend >=0.4.0,<0.5.0.dev0
     - scikit-build-core >=0.10.0
     - dlpack >=0.8,<1.0
-    - numba-cuda >=0.19.1,<0.20.0a0
+    - numba-cuda >=0.22.1,<0.23.0
     - libcudf =${{ version }}
     - pylibcudf =${{ version }}
     - rmm =${{ minor_version }}
@@ -86,7 +86,7 @@ requirements:
     - typing_extensions >=4.0.0
     - pandas >=2.0,<2.4.0dev0
     - cupy >=13.6.0
-    - numba-cuda >=0.19.1,<0.20.0a0
+    - numba-cuda >=0.22.1,<0.23.0
     - numba >=0.60.0,<0.62.0a0
     - numpy >=1.23,<3.0a0
     - pyarrow>=15.0.0,<22.0.0a0

dependencies.yaml

@@ -817,7 +817,7 @@ dependencies:
             packages:
               - numba==0.60.0
               - pandas==2.0.*
-              - numba-cuda==0.19.1
+              - numba-cuda==0.22.1
           - matrix: {dependencies: "latest"}
             packages:
               - pandas==2.3.3
@@ -960,18 +960,18 @@ dependencies:
     common:
       - output_types: [conda]
         packages:
-        - numba-cuda>=0.19.1,<0.20.0a0
+        - numba-cuda>=0.22.1,<0.23.0
     specific:
       - output_types: [requirements, pyproject]
         matrices:
           - matrix:
               cuda: "12.*"
             packages:
-              - numba-cuda[cu12]>=0.19.1,<0.20.0a0
+              - numba-cuda[cu12]>=0.22.1,<0.23.0
           # fallback to CUDA 13 versions if 'cuda' is '13.*' or not provided
           - matrix:
             packages:
-              - numba-cuda[cu13]>=0.19.1,<0.20.0a0
+              - numba-cuda[cu13]>=0.22.1,<0.23.0
   depends_on_pylibcudf:
     common:
       - output_types: conda

python/cudf/cudf/core/udf/masked_typing.py

@@ -5,7 +5,6 @@
 
 import numpy as np
 from numba import types
-from numba.core.datamodel import default_manager
 from numba.core.extending import (
     make_attribute_wrapper,
     models,
@@ -20,6 +19,7 @@
 )
 from numba.core.typing.typeof import typeof
 from numba.cuda.cudadecl import registry as cuda_decl_registry
+from numba.cuda.descriptor import cuda_target
 from numba.np.numpy_support import from_dtype
 
 from cudf.core.missing import NA
@@ -113,7 +113,9 @@ class MaskedType(types.Type):
     def __init__(self, value):
         # MaskedType in Numba shall be parameterized
         # with a value type
-        if default_manager[value].has_nrt_meminfo():
+        if cuda_target.target_context.data_model_manager[
+            value
+        ].has_nrt_meminfo():
             ctx = _current_nrt_context.get(None)
             if ctx is not None:
                 # we're in a compilation that is determining

python/cudf/cudf/core/udf/nrt_utils.py

@@ -4,7 +4,7 @@
 import contextvars
 from contextlib import contextmanager
 
-from numba import config as numba_config
+from numba.cuda import config as numba_config
 
 _current_nrt_context: contextvars.ContextVar = contextvars.ContextVar(
     "current_nrt_context"

python/cudf/cudf/core/udf/utils.py

@@ -14,8 +14,9 @@
 import numpy as np
 from cuda.bindings import runtime
 from numba import cuda, typeof
-from numba.core.datamodel import default_manager, models
+from numba.core.datamodel import models
 from numba.core.extending import register_model
+from numba.cuda.descriptor import cuda_target
 from numba.np import numpy_support
 from numba.types import CPointer, Record, Tuple, int64, void
 
@@ -303,7 +304,7 @@ def _get_extensionty_size(ty):
     Return the size of an extension type in bytes
     """
     target_data = ll.create_target_data(_nvvm_data_layout)
-    llty = default_manager[ty].get_value_type()
+    llty = cuda_target.target_context.data_model_manager[ty].get_value_type()
     return llty.get_abi_size(target_data)
 
 

python/cudf/cudf/tests/private_objects/test_nrt_stats.py

@@ -1,7 +1,7 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION.
 # SPDX-License-Identifier: Apache-2.0
 import pytest
-from numba import config
+from numba.cuda import config
 from numba.cuda.memory_management.nrt import rtsys
 
 import cudf

python/cudf/cudf/utils/_numba.py

@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 import numba
-from numba import config as numba_config
+from numba.cuda import config as numba_config
 from packaging import version
 
 

python/cudf/pyproject.toml

@@ -24,7 +24,7 @@ dependencies = [
     "cupy-cuda13x>=13.6.0",
     "fsspec>=0.6.0",
     "libcudf==26.2.*,>=0.0.0a0",
-    "numba-cuda[cu13]>=0.19.1,<0.20.0a0",
+    "numba-cuda[cu13]>=0.22.1,<0.23.0",
     "numba>=0.60.0,<0.62.0a0",
     "numpy>=1.23,<3.0a0",
     "nvtx>=0.2.1",
@@ -131,7 +131,7 @@ requires = [
     "libcudf==26.2.*,>=0.0.0a0",
     "librmm==26.2.*,>=0.0.0a0",
     "ninja",
-    "numba-cuda[cu13]>=0.19.1,<0.20.0a0",
+    "numba-cuda[cu13]>=0.22.1,<0.23.0",
     "pylibcudf==26.2.*,>=0.0.0a0",
     "rmm==26.2.*,>=0.0.0a0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.

python/pylibcudf/pyproject.toml

@@ -45,7 +45,7 @@ test = [
     "hypothesis>=6.131.7",
     "mmh3",
     "nanoarrow",
-    "numba-cuda[cu13]>=0.19.1,<0.20.0a0",
+    "numba-cuda[cu13]>=0.22.1,<0.23.0",
     "numba>=0.60.0,<0.62.0a0",
     "pandas",
     "pyarrow>=15.0.0,!=17.0.0; platform_machine=='aarch64'",