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

numba_cuda/numba/cuda/utils.py

@@ -28,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):
@@ -572,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-0.19.1.dist-info → numba_cuda-0.20.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: numba-cuda
-Version: 0.19.1
+Version: 0.20.0
 Summary: CUDA target for Numba
 Author: Anaconda Inc., NVIDIA Corporation
 License-Expression: BSD-2-Clause
@@ -39,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"