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

_numba_cuda_redirector.pth

@@ -1 +1,4 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import _numba_cuda_redirector

_numba_cuda_redirector.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import importlib
 import importlib.abc
 import pathlib

numba_cuda/VERSION

@@ -1 +1 @@
-0.18.1
+0.19.1

numba_cuda/__init__.py

@@ -1,4 +1,5 @@
-# Copyright (c) 2024, NVIDIA CORPORATION.
+# SPDX-FileCopyrightText: Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
 
 from numba_cuda._version import __version__
 

numba_cuda/_version.py

@@ -1,16 +1,5 @@
-# Copyright (c) 2024, NVIDIA CORPORATION.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# SPDX-FileCopyrightText: Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
 
 import importlib.resources
 

numba_cuda/numba/cuda/__init__.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import importlib
 from numba.core import config
 from .utils import _readenv
@@ -50,7 +53,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/numba/cuda/_internal/cuda_bf16.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 # Automatically generated by Numbast Static Binding Generator
 # Generator Information:
 # Ast_canopy version: 0.3.0
@@ -23,8 +26,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/numba/cuda/_internal/cuda_fp16.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 # Automatically generated by Numbast Static Binding Generator
 # Generator Information:
 # Ast_canopy version: 0.3.0
@@ -26,7 +29,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/numba/cuda/api.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 API that are reported to numba.cuda
 """
@@ -10,7 +13,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 +101,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/numba/cuda/api_util.py

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

numba_cuda/numba/cuda/args.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 Hints to wrap Kernel arguments to indicate how to manage host-device
 memory transfers before & after the kernel call.

numba_cuda/numba/cuda/bf16.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from numba.cuda._internal.cuda_bf16 import (
     _type_class___nv_bfloat16,
     nv_bfloat16 as bfloat16,

numba_cuda/numba/cuda/cg.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from numba.core import types
 from numba.core.extending import overload, overload_method
 from numba.core.typing import signature

numba_cuda/numba/cuda/cgutils.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 Generic helpers for LLVM code generation.
 """

numba_cuda/numba/cuda/codegen.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from llvmlite import ir
 
 from numba.core import config

numba_cuda/numba/cuda/compiler.py

@@ -1,14 +1,15 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 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 +23,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 +34,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

numba_cuda/numba/cuda/core/caching.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from abc import abstractmethod, ABCMeta
 import itertools
 import numba

numba_cuda/numba/cuda/core/callconv.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from numba.core import types
 from numba.cuda import cgutils
 from collections import namedtuple

numba_cuda/numba/cuda/core/codegen.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from abc import abstractmethod, ABCMeta
 from numba.misc.llvm_pass_timings import PassTimingsCollection
 

numba_cuda/numba/cuda/core/compiler.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from numba.core.tracing import event
 
 from numba.core import callconv, bytecode, config, errors