numba-cuda 0.19.0__py3-none-any.whl → 0.20.0__py3-none-any.whl

numba_cuda/numba/cuda/typing/templates.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 Define typing templates
 """
@@ -12,14 +15,15 @@ from collections.abc import Sequence
 from types import MethodType, FunctionType, MappingProxyType
 
 import numba
-from numba.core import types, utils, targetconfig
+from numba.core import types
 from numba.core.errors import (
     TypingError,
     InternalError,
 )
-from numba.core.cpu_options import InlineOptions
+from numba.cuda.core.options import InlineOptions
 from numba.core.typing.templates import Signature as CoreSignature
-from numba.cuda.core import ir_utils
+from numba.cuda import utils
+from numba.cuda.core import ir_utils, targetconfig
 
 # info store for inliner callback functions e.g. cost model
 _inline_info = namedtuple("inline_info", "func_ir typemap calltypes signature")
@@ -642,7 +646,7 @@ class _OverloadFunctionTemplate(AbstractTemplate):
         Type the overloaded function by compiling the appropriate
         implementation for the given args.
         """
-        from numba.core.typed_passes import PreLowerStripPhis
+        from numba.cuda.core.typed_passes import PreLowerStripPhis
 
         disp, new_args = self._get_impl(args, kws)
         if disp is None:
@@ -655,8 +659,8 @@ class _OverloadFunctionTemplate(AbstractTemplate):
         if not self._inline.is_never_inline:
             # need to run the compiler front end up to type inference to compute
             # a signature
-            from numba.core import typed_passes, compiler
-            from numba.core.inline_closurecall import InlineWorker
+            from numba.cuda.core import typed_passes, compiler
+            from numba.cuda.core.inline_closurecall import InlineWorker
 
             fcomp = disp._compiler
             flags = compiler.Flags()
@@ -1356,18 +1360,10 @@ class Registry(object):
         self.globals = []
 
     def register(self, item):
-        assert issubclass(
-            item,
-            (FunctionTemplate, numba.core.typing.templates.FunctionTemplate),
-        )
         self.functions.append(item)
         return item
 
     def register_attr(self, item):
-        assert issubclass(
-            item,
-            (AttributeTemplate, numba.core.typing.templates.AttributeTemplate),
-        )
         self.attributes.append(item)
         return item
 

numba_cuda/numba/cuda/ufuncs.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """Contains information on how to translate different ufuncs for the CUDA
 target. It is a database of different ufuncs and how each of its loops maps to
 a function that implements the inner kernel of that ufunc (the inner kernel
@@ -23,9 +26,9 @@ def get_ufunc_info(ufunc_key):
 @lru_cache
 def ufunc_db():
     # Imports here are at function scope to avoid circular imports
-    from numba.cpython import cmathimpl, mathimpl, numbers
-    from numba.np import npyfuncs
-    from numba.np.numpy_support import numpy_version
+    from numba.cuda.cpython import cmathimpl, mathimpl, numbers
+    from numba.cuda.np import npyfuncs
+    from numba.cuda.np.numpy_support import numpy_version
 
     def np_unary_impl(fn, context, builder, sig, args):
         npyfuncs._check_arity_and_homogeneity(sig, args, 1)

numba_cuda/numba/cuda/utils.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import os
 import warnings
 import traceback
@@ -25,13 +28,16 @@ from inspect import signature as pysignature  # noqa: F401
 from inspect import Signature as pySignature  # noqa: F401
 from inspect import Parameter as pyParameter  # noqa: F401
 
-from numba.core.config import (
+from numba.cuda.core.config import (
     MACHINE_BITS,  # noqa: F401
     DEVELOPER_MODE,
 )  # noqa: F401
-from numba.core import types, config
 
-from collections.abc import Mapping, Sequence, MutableSet, MutableMapping
+from numba.cuda.core import config
+
+from collections.abc import Mapping, MutableSet, MutableMapping
+
+PYVERSION = config.PYVERSION
 
 
 def erase_traceback(exc_value):
@@ -569,115 +575,6 @@ def get_nargs_range(pyfunc):
     return min_nargs, max_nargs
 
 
-def unify_function_types(numba_types):
-    """Return a normalized tuple of Numba function types so that
-
-        Tuple(numba_types)
-
-    becomes
-
-        UniTuple(dtype=<unified function type>, count=len(numba_types))
-
-    If the above transformation would be incorrect, return the
-    original input as given. For instance, if the input tuple contains
-    types that are not function or dispatcher type, the transformation
-    is considered incorrect.
-    """
-    dtype = unified_function_type(numba_types)
-    if dtype is None:
-        return numba_types
-    return (dtype,) * len(numba_types)
-
-
-def unified_function_type(numba_types, require_precise=True):
-    """Returns a unified Numba function type if possible.
-
-    Parameters
-    ----------
-    numba_types : Sequence of numba Type instances.
-    require_precise : bool
-      If True, the returned Numba function type must be precise.
-
-    Returns
-    -------
-    typ : {numba.core.types.Type, None}
-      A unified Numba function type. Or ``None`` when the Numba types
-      cannot be unified, e.g. when the ``numba_types`` contains at
-      least two different Numba function type instances.
-
-    If ``numba_types`` contains a Numba dispatcher type, the unified
-    Numba function type will be an imprecise ``UndefinedFunctionType``
-    instance, or None when ``require_precise=True`` is specified.
-
-    Specifying ``require_precise=False`` enables unifying imprecise
-    Numba dispatcher instances when used in tuples or if-then branches
-    when the precise Numba function cannot be determined on the first
-    occurrence that is not a call expression.
-    """
-    from numba.core.errors import NumbaExperimentalFeatureWarning
-
-    if not (
-        isinstance(numba_types, Sequence)
-        and len(numba_types) > 0
-        and isinstance(numba_types[0], (types.Dispatcher, types.FunctionType))
-    ):
-        return
-
-    warnings.warn(
-        "First-class function type feature is experimental",
-        category=NumbaExperimentalFeatureWarning,
-    )
-
-    mnargs, mxargs = None, None
-    dispatchers = set()
-    function = None
-    undefined_function = None
-
-    for t in numba_types:
-        if isinstance(t, types.Dispatcher):
-            mnargs1, mxargs1 = get_nargs_range(t.dispatcher.py_func)
-            if mnargs is None:
-                mnargs, mxargs = mnargs1, mxargs1
-            elif not (mnargs, mxargs) == (mnargs1, mxargs1):
-                return
-            dispatchers.add(t.dispatcher)
-            t = t.dispatcher.get_function_type()
-            if t is None:
-                continue
-        if isinstance(t, types.FunctionType):
-            if mnargs is None:
-                mnargs = mxargs = t.nargs
-            elif not (mnargs == mxargs == t.nargs):
-                return
-            if isinstance(t, types.UndefinedFunctionType):
-                if undefined_function is None:
-                    undefined_function = t
-                else:
-                    # Refuse to unify using function type
-                    return
-                dispatchers.update(t.dispatchers)
-            else:
-                if function is None:
-                    function = t
-                else:
-                    assert function == t
-        else:
-            return
-    if require_precise and (function is None or undefined_function is not None):
-        return
-    if function is not None:
-        if undefined_function is not None:
-            assert function.nargs == undefined_function.nargs
-            function = undefined_function
-    elif undefined_function is not None:
-        undefined_function.dispatchers.update(dispatchers)
-        function = undefined_function
-    else:
-        function = types.UndefinedFunctionType(mnargs, dispatchers)
-
-    return function
-
-
 class _RedirectSubpackage(ModuleType):
     """Redirect a subpackage to a subpackage.
 

numba_cuda/numba/cuda/vector_types.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 # CUDA built-in Vector Types
 # https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#built-in-vector-types
 

numba_cuda/numba/cuda/vectorizers.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from numba import cuda
 from numpy import array as np_array
 from numba.cuda import deviceufunc

{numba_cuda-0.19.0.dist-info → numba_cuda-0.20.0.dist-info}/METADATA

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: numba-cuda
-Version: 0.19.0
+Version: 0.20.0
 Summary: CUDA target for Numba
 Author: Anaconda Inc., NVIDIA Corporation
-License: BSD 2-clause
+License-Expression: BSD-2-Clause
 Project-URL: Homepage, https://nvidia.github.io/numba-cuda/
 Project-URL: Documentation, https://nvidia.github.io/numba-cuda/
 Project-URL: Repository, https://github.com/NVIDIA/numba-cuda
@@ -12,6 +12,7 @@ Project-URL: Issues, https://github.com/NVIDIA/numba-cuda/issues
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
+License-File: LICENSE.numba
 Requires-Dist: numba>=0.60.0
 Provides-Extra: cu12
 Requires-Dist: cuda-bindings<13.0.0,>=12.9.1; extra == "cu12"
@@ -21,6 +22,7 @@ Requires-Dist: nvidia-cuda-nvcc-cu12; extra == "cu12"
 Requires-Dist: nvidia-cuda-runtime-cu12; extra == "cu12"
 Requires-Dist: nvidia-cuda-nvrtc-cu12; extra == "cu12"
 Requires-Dist: nvidia-nvjitlink-cu12; extra == "cu12"
+Requires-Dist: nvidia-cuda-cccl-cu12; extra == "cu12"
 Provides-Extra: cu13
 Requires-Dist: cuda-bindings==13.*; extra == "cu13"
 Requires-Dist: cuda-core<0.4.0dev0,==0.3.2; extra == "cu13"
@@ -29,6 +31,7 @@ Requires-Dist: nvidia-nvvm==13.*; extra == "cu13"
 Requires-Dist: nvidia-cuda-runtime==13.*; extra == "cu13"
 Requires-Dist: nvidia-cuda-nvrtc==13.*; extra == "cu13"
 Requires-Dist: nvidia-nvjitlink==13.*; extra == "cu13"
+Requires-Dist: nvidia-cuda-cccl==13.*; extra == "cu13"
 Provides-Extra: test
 Requires-Dist: pre-commit; extra == "test"
 Requires-Dist: psutil; extra == "test"
@@ -36,6 +39,7 @@ Requires-Dist: cffi; extra == "test"
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: pytest-xdist; extra == "test"
 Requires-Dist: filecheck; extra == "test"
+Requires-Dist: ml_dtypes; extra == "test"
 Provides-Extra: test-cu12
 Requires-Dist: numba-cuda[cu12]; extra == "test-cu12"
 Requires-Dist: numba-cuda[test]; extra == "test-cu12"