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

numba_cuda/numba/cuda/simulator/__init__.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import sys
 
 from .api import *
@@ -17,7 +20,7 @@ from .cudadrv import devicearray
 from .cudadrv.devices import require_context, gpus
 from .cudadrv.devices import get_context as current_context
 from .cudadrv.runtime import runtime
-from numba.core import config
+from numba.cuda.core import config
 
 reduce = Reduce
 

numba_cuda/numba/cuda/simulator/_internal/__init__.py

@@ -1 +1,4 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from numba.cuda.simulator._internal import cuda_bf16  # noqa: F401

numba_cuda/numba/cuda/simulator/_internal/cuda_bf16.py

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

numba_cuda/numba/cuda/simulator/api.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 Contains CUDA API functions
 """
@@ -17,7 +20,7 @@ from .cudadrv.linkable_code import (
     LTOIR,  # noqa: F401
 )  # noqa: F401
 from .kernel import FakeCUDAKernel
-from numba.core import config
+from numba.cuda.core import config
 from numba.cuda.core.sigutils import is_signature
 from warnings import warn
 from ..args import In, Out, InOut  # noqa: F401

numba_cuda/numba/cuda/simulator/bf16.py

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

numba_cuda/numba/cuda/simulator/compiler.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 The compiler is not implemented in the simulator. This module provides a stub
 to allow tests to import successfully.
@@ -8,3 +11,7 @@ compile_for_current_device = None
 compile_ptx = None
 compile_ptx_for_current_device = None
 declare_device_function = None
+
+
+def run_frontend(func, inline_closures=False, emit_dels=False):
+    pass

numba_cuda/numba/cuda/simulator/cudadrv/__init__.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.simulator.cudadrv import (
     devicearray,
     devices,

numba_cuda/numba/cuda/simulator/cudadrv/devicearray.py

@@ -1,10 +1,13 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 The Device Array API is not implemented in the simulator. This module provides
 stubs to allow tests to import correctly.
 """
 
 from contextlib import contextmanager
-from numba.np.numpy_support import numpy_version
+from numba.cuda.np.numpy_support import numpy_version
 
 import numpy as np
 

numba_cuda/numba/cuda/simulator/cudadrv/devices.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
 from collections import namedtuple
 

numba_cuda/numba/cuda/simulator/cudadrv/driver.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 Most of the driver API is unsupported in the simulator, but some stubs are
 provided to allow tests to import correctly.

numba_cuda/numba/cuda/simulator/cudadrv/drvapi.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 drvapi is not implemented in the simulator, but this module exists to allow
 tests to import correctly.

numba_cuda/numba/cuda/simulator/cudadrv/dummyarray.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 # Dummy arrays are not implemented in the simulator. This file allows the dummy
 # array tests to be imported, but they are skipped on the simulator.
 

numba_cuda/numba/cuda/simulator/cudadrv/error.py

@@ -1,3 +1,7 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
+
 class CudaSupportError(RuntimeError):
     pass
 

numba_cuda/numba/cuda/simulator/cudadrv/libs.py

@@ -1,3 +1,7 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
+
 def check_static_lib(lib):
     raise FileNotFoundError("Linking libraries not supported by cudasim")
 

numba_cuda/numba/cuda/simulator/cudadrv/linkable_code.py

@@ -1,3 +1,7 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
+
 class LinkableCode:
     """An object that holds code to be linked from memory.
 

numba_cuda/numba/cuda/simulator/cudadrv/nvrtc.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 NVVM is not supported in the simulator, but stubs are provided to allow tests
 to import correctly.

numba_cuda/numba/cuda/simulator/cudadrv/nvvm.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 NVVM is not supported in the simulator, but stubs are provided to allow tests
 to import correctly.

numba_cuda/numba/cuda/simulator/cudadrv/runtime.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 The runtime API is unsupported in the simulator, but some stubs are
 provided to allow tests to import correctly.

numba_cuda/numba/cuda/simulator/dispatcher.py

@@ -1,3 +1,7 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
+
 class CUDADispatcher:
     """
     Dummy class so that consumers that try to import the real CUDADispatcher

numba_cuda/numba/cuda/simulator/kernel.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from contextlib import contextmanager
 import functools
 import sys

numba_cuda/numba/cuda/simulator/kernelapi.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 Implements the cuda module as called from within an executing kernel
 (@cuda.jit-decorated function).
@@ -10,7 +13,7 @@ import traceback
 from numba.core import types
 import numpy as np
 
-from numba.np import numpy_support
+from numba.cuda.np import numpy_support
 
 from .vector_types import vector_types
 

numba_cuda/numba/cuda/simulator/memory_management/__init__.py

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

numba_cuda/numba/cuda/simulator/memory_management/nrt.py

@@ -1,6 +1,21 @@
-from numba import config
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
 
-rtsys = None
+from numba.cuda import config
+
+
+class RTSys:
+    def __init__(self, *args, **kwargs):
+        pass
+
+    def memsys_enable_stats(self):
+        pass
+
+    def get_allocation_stats(self):
+        pass
+
+
+rtsys = RTSys()
 
 config.CUDA_NRT_STATS = False
 config.CUDA_ENABLE_NRT = False

numba_cuda/numba/cuda/simulator/reduction.py

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

numba_cuda/numba/cuda/simulator/vector_types.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 types
 from numba.cuda.stubs import _vector_type_stubs
 

numba_cuda/numba/cuda/simulator_init.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 # We import * from simulator here because * is imported from simulator_init by
 # numba.cuda.__init__.
 from .simulator import *  # noqa: F403, F401

numba_cuda/numba/cuda/stubs.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 This scripts specifies all PTX special objects.
 """

numba_cuda/numba/cuda/target.py

@@ -1,29 +1,36 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import re
 from functools import cached_property
 import llvmlite.binding as ll
 from llvmlite import ir
 import warnings
-from numba.cuda import cgutils, itanium_mangler
-from numba.core import (
-    compiler,
-    config,
-    targetconfig,
-    types,
-    typing,
-)
+
+from numba.core import types
+
 from numba.core.compiler_lock import global_compiler_lock
 from numba.core.dispatcher import Dispatcher
 from numba.core.errors import NumbaWarning
-from numba.core.base import BaseContext
-from numba.cuda.core.callconv import BaseCallConv, MinimalCallConv
+from numba.cuda.core.base import BaseContext
 from numba.core.typing import cmathdecl
 from numba.core import datamodel
 
 from .cudadrv import nvvm
-from numba.cuda import codegen, ufuncs
+from numba.cuda import (
+    cgutils,
+    itanium_mangler,
+    compiler,
+    codegen,
+    ufuncs,
+    typing,
+)
 from numba.cuda.debuginfo import CUDADIBuilder
 from numba.cuda.flags import CUDAFlags
 from numba.cuda.models import cuda_data_manager
+from numba.cuda.core.callconv import BaseCallConv, MinimalCallConv
+from numba.cuda.core import config, targetconfig
+
 
 # -----------------------------------------------------------------------------
 # Typing
@@ -31,8 +38,15 @@ from numba.cuda.models import cuda_data_manager
 
 class CUDATypingContext(typing.BaseContext):
     def load_additional_registries(self):
-        from . import cudadecl, cudamath, fp16, libdevicedecl, vector_types
-        from numba.core.typing import enumdecl, cffi_utils
+        from . import (
+            cudadecl,
+            cudamath,
+            fp16,
+            bf16,
+            libdevicedecl,
+            vector_types,
+        )
+        from numba.cuda.typing import enumdecl, cffi_utils
 
         self.install_registry(cudadecl.registry)
         self.install_registry(cffi_utils.registry)
@@ -42,6 +56,7 @@ class CUDATypingContext(typing.BaseContext):
         self.install_registry(enumdecl.registry)
         self.install_registry(vector_types.typing_registry)
         self.install_registry(fp16.typing_registry)
+        self.install_registry(bf16.typing_registry)
 
     def resolve_value_type(self, val):
         # treat other dispatcher object as another device function
@@ -138,12 +153,14 @@ class CUDATargetContext(BaseContext):
         self._target_data = None
 
     def load_additional_registries(self):
-        # side effect of import needed for numba.cpython.*, the builtins
+        # side effect of import needed for numba.cpython.*, numba.cuda.cpython.*, the builtins
         # registry is updated at import time.
-        from numba.cpython import numbers, tupleobj, slicing  # noqa: F401
+        from numba.cpython import tupleobj, slicing  # noqa: F401
+        from numba.cuda.cpython import numbers  # noqa: F401
         from numba.cpython import rangeobj, iterators, enumimpl  # noqa: F401
         from numba.cpython import unicode, charseq  # noqa: F401
-        from numba.cpython import cmathimpl
+        from numba.cuda.cpython import cmathimpl, mathimpl
+        from numba.core import optional  # noqa: F401
         from numba.misc import cffiimpl
         from numba.np import arrayobj  # noqa: F401
         from numba.np import npdatetime  # noqa: F401
@@ -152,8 +169,9 @@ class CUDATargetContext(BaseContext):
             fp16,
             printimpl,
             libdeviceimpl,
-            mathimpl,
+            mathimpl as cuda_mathimpl,
             vector_types,
+            bf16,
         )
 
         # fix for #8940
@@ -165,8 +183,11 @@ class CUDATargetContext(BaseContext):
         self.install_registry(libdeviceimpl.registry)
         self.install_registry(cmathimpl.registry)
         self.install_registry(mathimpl.registry)
+        self.install_registry(numbers.registry)
+        self.install_registry(cuda_mathimpl.registry)
         self.install_registry(vector_types.impl_registry)
         self.install_registry(fp16.target_registry)
+        self.install_registry(bf16.target_registry)
 
     def codegen(self):
         return self._internal_codegen

numba_cuda/numba/cuda/testing.py

@@ -1,13 +1,16 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import os
 import platform
 import shutil
 import pytest
 from datetime import datetime
-from numba.core.utils import PYVERSION
-from numba.cuda.cuda_paths import get_conda_ctk
+from numba.cuda.utils import PYVERSION
+from numba.cuda.cuda_paths import get_conda_ctk_libdir
 from numba.cuda.cudadrv import driver, devices, libs
 from numba.cuda.dispatcher import CUDADispatcher
-from numba.core import config
+from numba.cuda import config
 from numba.cuda.tests.support import TestCase
 from pathlib import Path
 
@@ -208,7 +211,7 @@ def skip_unless_cudasim(reason):
 
 def skip_unless_conda_cudatoolkit(reason):
     """Skip test if the CUDA toolkit was not installed by Conda"""
-    return unittest.skipUnless(get_conda_ctk() is not None, reason)
+    return unittest.skipUnless(get_conda_ctk_libdir() is not None, reason)
 
 
 def skip_if_external_memmgr(reason):
@@ -269,21 +272,6 @@ def skip_if_mvc_enabled(reason):
     )
 
 
-def skip_if_mvc_libraries_unavailable(fn):
-    libs_available = False
-    try:
-        import cubinlinker  # noqa: F401 # type: ignore
-        import ptxcompiler  # noqa: F401 # type: ignore
-
-        libs_available = True
-    except ImportError:
-        pass
-
-    return unittest.skipUnless(
-        libs_available, "Requires cubinlinker and ptxcompiler"
-    )(fn)
-
-
 def cc_X_or_above(major, minor):
     if not config.ENABLE_CUDASIM:
         cc = devices.get_context().device.compute_capability

numba_cuda/numba/cuda/tests/__init__.py

@@ -1,5 +1,8 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from fnmatch import fnmatch
-from numba.testing import unittest
+import unittest
 from numba import cuda
 from os.path import dirname, isfile, join, normpath, relpath, splitext
 

numba_cuda/numba/cuda/tests/cloudpickle_main_class.py

@@ -0,0 +1,9 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
+# Expected to run this module as __main__
+
+
+# Cloudpickle will think this is a dynamic class when this module is __main__
+class Klass:
+    classvar = None

numba_cuda/numba/cuda/tests/complex_usecases.py

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

numba_cuda/numba/cuda/tests/core/serialize_usecases.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 Separate module with function samples for serialization tests,
 to avoid issues with __main__.

numba_cuda/numba/cuda/tests/core/test_itanium_mangler.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 # -*- coding: utf-8 -*-
 from numba import int32, int64, uint32, uint64, float32, float64
 from numba.core.types import range_iter32_type

numba_cuda/numba/cuda/tests/core/test_serialize.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import contextlib
 import gc
 import pickle
@@ -9,9 +12,9 @@ from multiprocessing import get_context
 
 import numba
 from numba.core.errors import TypingError
-from numba.tests.support import TestCase
+from numba.cuda.tests.support import TestCase
 from numba.core.target_extension import resolve_dispatcher_from_str
-from numba.cloudpickle import dumps, loads
+from numba.cuda.cloudpickle import dumps, loads
 
 
 class TestDispatcherPickling(TestCase):
@@ -227,7 +230,7 @@ class TestDispatcherPickling(TestCase):
 class TestSerializationMisc(TestCase):
     def test_numba_unpickle(self):
         # Test that _numba_unpickle is memorizing its output
-        from numba.core.serialize import _numba_unpickle
+        from numba.cuda.serialize import _numba_unpickle
 
         random_obj = object()
         bytebuf = pickle.dumps(random_obj)
@@ -315,7 +318,7 @@ def issue_7356():
 def check_main_class_reset_on_unpickle():
     # Load module and get its global dictionary
     glbs = runpy.run_module(
-        "numba.tests.cloudpickle_main_class",
+        "numba.cuda.tests.cloudpickle_main_class",
         run_name="__main__",
     )
     # Get the Klass and check it is from __main__

numba_cuda/numba/cuda/tests/cudadrv/__init__.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.tests import load_testsuite
 import os
 

numba_cuda/numba/cuda/tests/cudadrv/test_array_attr.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
 from numba import cuda
 from numba.cuda.testing import unittest, CUDATestCase, skip_on_cudasim

numba_cuda/numba/cuda/tests/cudadrv/test_context_stack.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import numbers
 from ctypes import byref
 import weakref

numba_cuda/numba/cuda/tests/cudadrv/test_cuda_array_slicing.py

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

numba_cuda/numba/cuda/tests/cudadrv/test_cuda_auto_context.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
 from numba import cuda
 from numba.cuda.testing import unittest, CUDATestCase

numba_cuda/numba/cuda/tests/cudadrv/test_cuda_devicerecord.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
 import ctypes
 from numba.cuda.cudadrv.devicearray import (
@@ -7,7 +10,7 @@ from numba.cuda.cudadrv.devicearray import (
 )
 from numba.cuda.testing import unittest, CUDATestCase
 from numba.cuda.testing import skip_on_cudasim
-from numba.np import numpy_support
+from numba.cuda.np import numpy_support
 from numba import cuda
 
 N_CHARS = 5

numba_cuda/numba/cuda/tests/cudadrv/test_cuda_driver.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from ctypes import byref, c_int, sizeof
 
 from numba.cuda.cudadrv.driver import (

numba_cuda/numba/cuda/tests/cudadrv/test_cuda_libraries.py

@@ -1,6 +1,9 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from numba.cuda.testing import unittest
 from numba.cuda.testing import skip_on_cudasim, skip_unless_conda_cudatoolkit
-from numba.misc.findlib import find_lib
+from numba.cuda.misc.findlib import find_lib
 
 
 @skip_on_cudasim("Library detection unsupported in the simulator")

numba_cuda/numba/cuda/tests/cudadrv/test_cuda_memory.py

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

numba_cuda/numba/cuda/tests/cudadrv/test_cuda_ndarray.py

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

numba_cuda/numba/cuda/tests/cudadrv/test_deallocations.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from contextlib import contextmanager
 
 import numpy as np
@@ -10,7 +13,7 @@ from numba.cuda.testing import (
     CUDATestCase,
 )
 from numba.cuda.tests.support import captured_stderr
-from numba.core import config
+from numba.cuda.core import config
 
 
 @skip_on_cudasim("not supported on CUDASIM")