numba-cuda 0.18.1__tar.gz → 0.19.0__tar.gz

{numba_cuda-0.18.1 → numba_cuda-0.19.0}/PKG-INFO

@@ -1,56 +1,49 @@
 Metadata-Version: 2.4
 Name: numba-cuda
-Version: 0.18.1
+Version: 0.19.0
 Summary: CUDA target for Numba
 Author: Anaconda Inc., NVIDIA Corporation
 License: BSD 2-clause
-Project-URL: Homepage, https://github.com/rapidsai/numba-cuda
-Project-URL: Documentation, https://github.com/rapidsai/numba-cuda/blob/main/README.md
-Project-URL: Repository, https://github.com/rapidsai/numba-cuda
-Project-URL: License, https://github.com/rapidsai/numba-cuda/blob/main/LICENSE
+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
+Project-URL: License, https://github.com/NVIDIA/numba-cuda/blob/main/LICENSE
+Project-URL: Issues, https://github.com/NVIDIA/numba-cuda/issues
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: numba>=0.60.0
-Provides-Extra: cu11
-Requires-Dist: cuda-bindings==11.8.*; extra == "cu11"
-Requires-Dist: cuda-core==0.3.*; extra == "cu11"
-Requires-Dist: cuda-python==11.8.*; extra == "cu11"
-Requires-Dist: nvidia-cuda-nvcc-cu11; extra == "cu11"
-Requires-Dist: nvidia-cuda-runtime-cu11; extra == "cu11"
-Requires-Dist: nvidia-cuda-nvrtc-cu11; extra == "cu11"
 Provides-Extra: cu12
-Requires-Dist: cuda-bindings==12.9.*; extra == "cu12"
+Requires-Dist: cuda-bindings<13.0.0,>=12.9.1; extra == "cu12"
 Requires-Dist: cuda-core==0.3.*; extra == "cu12"
 Requires-Dist: cuda-python==12.9.*; extra == "cu12"
 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"
+Provides-Extra: cu13
+Requires-Dist: cuda-bindings==13.*; extra == "cu13"
+Requires-Dist: cuda-core<0.4.0dev0,==0.3.2; extra == "cu13"
+Requires-Dist: cuda-python==13.*; extra == "cu13"
+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"
 Provides-Extra: test
+Requires-Dist: pre-commit; extra == "test"
 Requires-Dist: psutil; extra == "test"
 Requires-Dist: cffi; extra == "test"
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: pytest-xdist; extra == "test"
 Requires-Dist: filecheck; extra == "test"
-Provides-Extra: test-cu11
-Requires-Dist: numba-cuda[test]; extra == "test-cu11"
-Requires-Dist: nvidia-curand-cu11; extra == "test-cu11"
 Provides-Extra: test-cu12
+Requires-Dist: numba-cuda[cu12]; extra == "test-cu12"
 Requires-Dist: numba-cuda[test]; extra == "test-cu12"
 Requires-Dist: nvidia-curand-cu12; extra == "test-cu12"
-Requires-Dist: pynvjitlink-cu12; extra == "test-cu12"
-Provides-Extra: dev
-Requires-Dist: pre-commit; extra == "dev"
-Requires-Dist: numba-cuda[test]; extra == "dev"
-Provides-Extra: dev-cu11
-Requires-Dist: numba-cuda[test-cu11]; extra == "dev-cu11"
-Requires-Dist: numba-cuda[test]; extra == "dev-cu11"
-Requires-Dist: numba-cuda[dev]; extra == "dev-cu11"
-Provides-Extra: dev-cu12
-Requires-Dist: numba-cuda[test-cu12]; extra == "dev-cu12"
-Requires-Dist: numba-cuda[test]; extra == "dev-cu12"
-Requires-Dist: numba-cuda[dev]; extra == "dev-cu12"
+Provides-Extra: test-cu13
+Requires-Dist: numba-cuda[cu13]; extra == "test-cu13"
+Requires-Dist: numba-cuda[test]; extra == "test-cu13"
+Requires-Dist: nvidia-curand==10.4.*; extra == "test-cu13"
 Dynamic: license-file
 
 <div align="center"><img src="docs/source/_static/numba-green-icon-rgb.svg" width="200"/></div>

numba_cuda-0.19.0/numba_cuda/VERSION

@@ -0,0 +1 @@
+0.19.0

{numba_cuda-0.18.1 → numba_cuda-0.19.0}/numba_cuda/numba/cuda/__init__.py

@@ -50,7 +50,7 @@ if config.CUDA_USE_NVIDIA_BINDING:
         )
 
 if config.CUDA_ENABLE_PYNVJITLINK:
-    if USE_NV_BINDING:
+    if USE_NV_BINDING and not pynvjitlink_auto_enabled:
         warnings.warn(
             "Explicitly enabling pynvjitlink is no longer necessary. "
             "NVIDIA bindings are enabled. cuda.core will be used "

{numba_cuda-0.18.1 → numba_cuda-0.19.0}/numba_cuda/numba/cuda/_internal/cuda_bf16.py

@@ -23,8 +23,8 @@ from numba.core.extending import (
     make_attribute_wrapper,
     register_model,
 )
-from numba.core.typing import signature
-from numba.core.typing.templates import AttributeTemplate, ConcreteTemplate
+from numba.cuda.typing import signature
+from numba.cuda.typing.templates import AttributeTemplate, ConcreteTemplate
 from numba.cuda import CUSource, declare_device
 from numba.cuda.cudadecl import register, register_attr, register_global
 from numba.cuda.cudaimpl import lower

{numba_cuda-0.18.1 → numba_cuda-0.19.0}/numba_cuda/numba/cuda/_internal/cuda_fp16.py

@@ -26,7 +26,7 @@ from numba.core.extending import (
 from numba.core.imputils import Registry as TargetRegistry, lower_cast
 import numba.core.typeconv
 from numba.core.typing import signature
-from numba.core.typing.templates import (
+from numba.cuda.typing.templates import (
     Registry as TypingRegistry,
     AbstractTemplate,
     AttributeTemplate,

{numba_cuda-0.18.1 → numba_cuda-0.19.0}/numba_cuda/numba/cuda/api.py

@@ -10,7 +10,6 @@ import numpy as np
 from .cudadrv import devicearray, devices, driver
 from numba.core import config
 from numba.cuda.api_util import prepare_shape_strides_dtype
-from numba.cuda.cudadrv.runtime import get_version
 
 # NDarray device helper
 
@@ -99,13 +98,9 @@ def is_float16_supported():
 def is_bfloat16_supported():
     """Whether bfloat16 are supported.
 
-    bfloat16 are only supported on devices with compute capability >= 8.0 and cuda version >= 12.0
+    bfloat16 is only supported on devices with compute capability >= 8.0
     """
-    cuda_version = get_version()
-    return current_context().device.supports_bfloat16 and cuda_version >= (
-        12,
-        0,
-    )
+    return current_context().device.supports_bfloat16
 
 
 @require_context

{numba_cuda-0.18.1 → numba_cuda-0.19.0}/numba_cuda/numba/cuda/compiler.py

@@ -1,14 +1,12 @@
 from llvmlite import ir
 from collections import namedtuple
 from numba.core import ir as numba_ir
-from numba.cuda import cgutils
+from numba.cuda import cgutils, typing
 from numba.core import (
     types,
-    typing,
     funcdesc,
     config,
     compiler,
-    sigutils,
 )
 from numba.core.compiler import (
     sanitize_compile_result_entries,
@@ -22,7 +20,6 @@ from numba.core.compiler_machinery import (
     PassManager,
     register_pass,
 )
-from numba.core.interpreter import Interpreter
 from numba.core.errors import NumbaInvalidConfigWarning
 from numba.core.untyped_passes import TranslateByteCode
 from numba.core.typed_passes import (
@@ -34,12 +31,18 @@ from numba.cuda import nvvmutils
 from numba.cuda.api import get_current_device
 from numba.cuda.codegen import ExternalCodeLibrary
 from numba.cuda.core.typed_passes import BaseNativeLowering
+from numba.cuda.core import sigutils
 from numba.cuda.cudadrv import nvvm, nvrtc
 from numba.cuda.descriptor import cuda_target
 from numba.cuda.flags import CUDAFlags
 from numba.cuda.target import CUDACABICallConv
 from numba.cuda import lowering, utils
+from numba.core.utils import PYVERSION
 
+if PYVERSION < (3, 10):
+    from numba.core.interpreter import Interpreter
+else:
+    from numba.cuda.core.interpreter import Interpreter
 
 # The CUDACompileResult (CCR) has a specially-defined entry point equal to its
 # id.  This is because the entry point is used as a key into a dict of