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

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,3 +1,6 @@
+# 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.

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,10 +1,11 @@
+# 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.
 """
 
-from numba import config
-
 
 def device_memset(dst, val, size, stream=0):
     dst.view("u1")[:size].fill(bytes([val])[0])
@@ -63,11 +64,6 @@ def launch_kernel(*args, **kwargs):
 
 USE_NV_BINDING = False
 
-PyNvJitLinker = None
-
-if config.ENABLE_CUDASIM:
-    config.CUDA_ENABLE_PYNVJITLINK = False
-
 
 def _have_nvjitlink():
     return False

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).

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,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from numba import config
 
 rtsys = None

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,13 +1,15 @@
+# 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
+from numba.cuda import cgutils, itanium_mangler
 from numba.core import (
     compiler,
     config,
-    itanium_mangler,
     targetconfig,
     types,
     typing,

numba_cuda/numba/cuda/testing.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 platform
 import shutil
@@ -8,7 +11,7 @@ from numba.cuda.cuda_paths import get_conda_ctk
 from numba.cuda.cudadrv import driver, devices, libs
 from numba.cuda.dispatcher import CUDADispatcher
 from numba.core import config
-from numba.tests.support import TestCase
+from numba.cuda.tests.support import TestCase
 from pathlib import Path
 
 from typing import Iterable, Union
@@ -154,7 +157,6 @@ class CUDATestCase(TestCase):
         matcher.stderr = StringIO()
         result = matcher.run()
         if result != 0:
-            dump_instructions = ""
             if self._dump_failed_filechecks:
                 dump_directory = Path(
                     datetime.now().strftime("numba-ir-%Y_%m_%d_%H_%M_%S")
@@ -172,13 +174,12 @@ class CUDATestCase(TestCase):
                 ):
                     _ = ir_file.write(ir_content + "\n")
                     _ = checks_file.write(check_patterns)
-                    dump_instructions = f"Reproduce with:\n\nfilecheck --check-prefixes={','.join(check_prefixes)} {checks_dump} --input-file={ir_dump}"
+                    dump_instructions = f"Reproduce with:\n\nfilecheck --check-prefixes={','.join(check_prefixes)} {checks_dump} --input-file {ir_dump}"
+            else:
+                dump_instructions = "Rerun with --dump-failed-filechecks to generate a reproducer."
 
             self.fail(
                 f"FileCheck failed:\n{matcher.stderr.getvalue()}\n\n"
-                + f"Check prefixes:\n{check_prefixes}\n\n"
-                + f"Check patterns:\n{check_patterns}\n"
-                + f"IR:\n{ir_content}\n\n"
                 + dump_instructions
             )
 

numba_cuda/numba/cuda/tests/__init__.py

@@ -1,3 +1,6 @@
+# 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
 from numba import cuda

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

@@ -0,0 +1,83 @@
+# 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
+from numba.cuda import itanium_mangler
+import unittest
+
+
+class TestItaniumManager(unittest.TestCase):
+    def test_ident(self):
+        got = itanium_mangler.mangle_identifier("apple")
+        expect = "5apple"
+        self.assertEqual(expect, got)
+
+        got = itanium_mangler.mangle_identifier("ap_ple")
+        expect = "6ap_ple"
+        self.assertEqual(expect, got)
+
+        got = itanium_mangler.mangle_identifier("apple213")
+        expect = "8apple213"
+        self.assertEqual(expect, got)
+
+    def test_types(self):
+        got = itanium_mangler.mangle_type(int32)
+        expect = "i"
+        self.assertEqual(expect, got)
+
+        got = itanium_mangler.mangle_type(int64)
+        expect = "x"
+        self.assertEqual(expect, got)
+
+        got = itanium_mangler.mangle_type(uint32)
+        expect = "j"
+        self.assertEqual(expect, got)
+
+        got = itanium_mangler.mangle_type(uint64)
+        expect = "y"
+        self.assertEqual(expect, got)
+
+        got = itanium_mangler.mangle_type(float32)
+        expect = "f"
+        self.assertEqual(expect, got)
+
+        got = itanium_mangler.mangle_type(float64)
+        expect = "d"
+        self.assertEqual(expect, got)
+
+    def test_function(self):
+        got = itanium_mangler.mangle("what", [int32, float32])
+        expect = "_Z4whatif"
+        self.assertEqual(expect, got)
+
+        got = itanium_mangler.mangle(
+            "a_little_brown_fox", [uint64, uint32, float64]
+        )
+        expect = "_Z18a_little_brown_foxyjd"
+        self.assertEqual(expect, got)
+
+    def test_custom_type(self):
+        got = itanium_mangler.mangle_type(range_iter32_type)
+        name = str(range_iter32_type)
+        expect = "{n}{name}".format(n=len(name), name=name)
+        self.assertEqual(expect, got)
+
+    def test_mangle_literal(self):
+        # check int
+        got = itanium_mangler.mangle_value(123)
+        expect = "Li123E"
+        self.assertEqual(expect, got)
+        # check float (not handled using standard)
+        got = itanium_mangler.mangle_value(12.3)
+        self.assertRegex(got, r"^\d+_12_[0-9a-z][0-9a-z]3$")
+
+    def test_mangle_unicode(self):
+        name = "f∂ƒ©z"
+        got = itanium_mangler.mangle_identifier(name)
+        self.assertRegex(got, r"^\d+f(_[a-z0-9][a-z0-9])+z$")
+
+
+if __name__ == "__main__":
+    unittest.main()

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

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 (

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,3 +1,6 @@
+# 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

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,10 +1,13 @@
+# 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
 from numba import cuda
 from numba.cuda.testing import unittest, CUDATestCase
 from numba.cuda.testing import skip_on_cudasim
-from numba.tests.support import IS_NUMPY_2
+from numba.cuda.tests.support import IS_NUMPY_2
 
 
 class TestCudaNDArray(CUDATestCase):

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
@@ -9,7 +12,7 @@ from numba.cuda.testing import (
     skip_if_external_memmgr,
     CUDATestCase,
 )
-from numba.tests.support import captured_stderr
+from numba.cuda.tests.support import captured_stderr
 from numba.core import config
 
 

numba_cuda/numba/cuda/tests/cudadrv/test_detect.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 sys
 import subprocess
@@ -9,7 +12,7 @@ from numba.cuda.testing import (
     skip_on_cudasim,
     skip_under_cuda_memcheck,
 )
-from numba.tests.support import captured_stdout
+from numba.cuda.tests.support import captured_stdout
 
 
 class TestCudaDetect(CUDATestCase):

numba_cuda/numba/cuda/tests/cudadrv/test_emm_plugins.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
 import weakref
@@ -5,7 +8,7 @@ import weakref
 from numba import cuda
 from numba.core import config
 from numba.cuda.testing import unittest, CUDATestCase, skip_on_cudasim
-from numba.tests.support import linux_only
+from numba.cuda.tests.support import linux_only
 
 if not config.ENABLE_CUDASIM:
 

numba_cuda/numba/cuda/tests/cudadrv/test_events.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_host_alloc.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.cuda.cudadrv import driver
 from numba import cuda

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

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

numba_cuda/numba/cuda/tests/cudadrv/test_inline_ptx.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.cuda.cudadrv import nvvm

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

numba_cuda/numba/cuda/tests/cudadrv/test_linker.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 warnings
 from numba import config
@@ -6,7 +9,7 @@ from numba.cuda.testing import skip_on_cudasim, skip_if_cuda_includes_missing
 from numba.cuda.testing import CUDATestCase, test_data_dir
 from numba.cuda.cudadrv.driver import CudaAPIError, _Linker, LinkerError
 from numba.cuda import require_context
-from numba.tests.support import ignore_internal_warnings
+from numba.cuda.tests.support import ignore_internal_warnings
 from numba import cuda, void, float64, int64, int32, typeof, float32
 from numba.cuda.cudadrv.error import NvrtcError
 

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

@@ -1,10 +1,13 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import numpy as np
 from ctypes import byref, c_size_t
 from numba.cuda.cudadrv.driver import device_memset, driver, USE_NV_BINDING
 from numba import cuda
 from numba.cuda.testing import unittest, ContextResettingTestCase
 from numba.cuda.testing import skip_on_cudasim, skip_on_arm
-from numba.tests.support import linux_only
+from numba.cuda.tests.support import linux_only
 
 
 @skip_on_cudasim("CUDA Driver API unsupported in the simulator")

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

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

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

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import multiprocessing as mp
 import traceback
 from numba.cuda.testing import unittest, CUDATestCase
@@ -6,7 +9,7 @@ from numba.cuda.testing import (
     skip_under_cuda_memcheck,
     skip_if_mvc_libraries_unavailable,
 )
-from numba.tests.support import linux_only
+from numba.cuda.tests.support import linux_only
 
 
 def child_test():

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

numba_cuda/numba/cuda/tests/cudadrv/test_nvrtc.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.cudadrv import nvrtc
 from numba.cuda.testing import skip_on_cudasim
 
@@ -13,13 +16,11 @@ class TestArchOption(unittest.TestCase):
         self.assertEqual(nvrtc.get_arch_option(8, 5), "compute_80")
         self.assertEqual(nvrtc.get_arch_option(9, 1), "compute_90")
         # Test known arch.
-        supported_cc = nvrtc.NVRTC().get_supported_archs()
-        for arch in supported_cc:
-            self.assertEqual(
-                nvrtc.get_arch_option(*arch), "compute_%d%d" % arch
-            )
+        supported_ccs = nvrtc.get_supported_ccs()
+        for cc in supported_ccs:
+            self.assertEqual(nvrtc.get_arch_option(*cc), "compute_%d%d" % cc)
         self.assertEqual(
-            nvrtc.get_arch_option(1000, 0), "compute_%d%d" % supported_cc[-1]
+            nvrtc.get_arch_option(1000, 0), "compute_%d%d" % supported_ccs[-1]
         )