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

numba_cuda/numba/cuda/lowering.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from collections import namedtuple, defaultdict
 import operator
 import warnings
@@ -1841,8 +1844,7 @@ class CUDALower(Lower):
             int_type = (llvm_ir.IntType,)
             real_type = llvm_ir.FloatType, llvm_ir.DoubleType
             if isinstance(lltype, int_type + real_type):
-                index = name.find(".")
-                src_name = name[:index] if index > 0 else name
+                src_name = name.split(".")[0]
                 if src_name in self.poly_var_typ_map:
                     # Do not emit debug value on polymorphic type var
                     return
@@ -1869,6 +1871,9 @@ class CUDALower(Lower):
 
         self.poly_var_typ_map = {}
         self.poly_var_loc_map = {}
+        self.poly_var_set = set()
+        self.poly_cleaned = False
+        self.lastblk = max(self.blocks.keys())
 
         # When debug info is enabled, walk through function body and mark
         # variables with polymorphic types.
@@ -1880,8 +1885,7 @@ class CUDALower(Lower):
                     if x.target.name.startswith("$"):
                         continue
                     ssa_name = x.target.name
-                    index = ssa_name.find(".")
-                    src_name = ssa_name[:index] if index > 0 else ssa_name
+                    src_name = ssa_name.split(".")[0]
                     # Check all the multi-versioned targets
                     if len(x.target.versioned_names) > 0:
                         fetype = self.typeof(ssa_name)
@@ -1902,12 +1906,12 @@ class CUDALower(Lower):
         """
         # If the name is not handled yet and a store is needed
         if name not in self.varmap and self.store_var_needed(name):
-            index = name.find(".")
-            src_name = name[:index] if index > 0 else name
+            src_name = name.split(".")[0]
             if src_name in self.poly_var_typ_map:
-                dtype = types.UnionType(self.poly_var_typ_map[src_name])
-                datamodel = self.context.data_model_manager[dtype]
+                self.poly_var_set.add(name)
                 if src_name not in self.poly_var_loc_map:
+                    dtype = types.UnionType(self.poly_var_typ_map[src_name])
+                    datamodel = self.context.data_model_manager[dtype]
                     # UnionType has sorted set of types, max at last index
                     maxsizetype = dtype.types[-1]
                     # Create a single element aggregate type
@@ -1916,13 +1920,7 @@ class CUDALower(Lower):
                     ptr = self.alloca_lltype(src_name, lltype, datamodel)
                     # save the location of the union type for polymorphic var
                     self.poly_var_loc_map[src_name] = ptr
-                # Any member of this union type shoud type cast ptr to fetype
-                lltype = self.context.get_value_type(fetype)
-                castptr = self.builder.bitcast(
-                    self.poly_var_loc_map[src_name], llvm_ir.PointerType(lltype)
-                )
-                # Remember the pointer
-                self.varmap[name] = castptr
+                return
 
         super()._alloca_var(name, fetype)
 
@@ -1936,6 +1934,45 @@ class CUDALower(Lower):
             or self._disable_sroa_like_opt
         )
 
+    def delvar(self, name):
+        """
+        Delete the given variable.
+        """
+        if name in self.poly_var_set:
+            fetype = self.typeof(name)
+            src_name = name.split(".")[0]
+            ptr = self.poly_var_loc_map[src_name]
+            self.decref(fetype, self.builder.load(ptr))
+            if (
+                self._cur_ir_block == self.blocks[self.lastblk]
+                and not self.poly_cleaned
+            ):
+                # Zero-fill the debug union for polymorphic only
+                # at the last block
+                for v in self.poly_var_loc_map.values():
+                    self.builder.store(
+                        llvm_ir.Constant(v.type.pointee, None), v
+                    )
+                    self.poly_cleaned = True
+            return
+
+        super().delvar(name)
+
+    def getvar(self, name):
+        """
+        Get a pointer to the given variable's slot.
+        """
+        if name in self.poly_var_set:
+            src_name = name.split(".")[0]
+            fetype = self.typeof(name)
+            lltype = self.context.get_value_type(fetype)
+            castptr = self.builder.bitcast(
+                self.poly_var_loc_map[src_name], llvm_ir.PointerType(lltype)
+            )
+            return castptr
+        else:
+            return super().getvar(name)
+
 
 def _lit_or_omitted(value):
     """Returns a Literal instance if the type of value is supported;

numba_cuda/numba/cuda/mathimpl.py

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

numba_cuda/numba/cuda/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 numba.cuda.memory_management.nrt import rtsys  # noqa: F401

numba_cuda/numba/cuda/memory_management/memsys.cu

@@ -1,3 +1,8 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
 #include "memsys.cuh"
 
 __device__ size_t memsys_size = sizeof(NRT_MemSys);

numba_cuda/numba/cuda/memory_management/memsys.cuh

@@ -1,3 +1,8 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
 #include <cuda/atomic>
 
 // Globally needed variables

numba_cuda/numba/cuda/memory_management/nrt.cu

@@ -1,3 +1,8 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
 #ifndef _NRT_H
 #define _NRT_H
 

numba_cuda/numba/cuda/memory_management/nrt.cuh

@@ -1,3 +1,8 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
 #include <cuda/atomic>
 
 typedef void (*NRT_dtor_function)(void* ptr, size_t size, void* info);

numba_cuda/numba/cuda/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
+
 import ctypes
 import os
 from functools import wraps
@@ -10,6 +13,7 @@ from numba.cuda.cudadrv.driver import (
     driver,
     launch_kernel,
     USE_NV_BINDING,
+    _have_nvjitlink,
 )
 from numba.cuda.cudadrv import devices
 from numba.cuda.api import get_current_device
@@ -80,7 +84,7 @@ class _Runtime:
         cc = get_current_device().compute_capability
 
         # Create a new linker instance and add the cu file
-        linker = _Linker.new(cc=cc)
+        linker = _Linker.new(cc=cc, lto=_have_nvjitlink())
         linker.add_cu_file(memsys_mod)
 
         # Complete the linker and create a module from it

numba_cuda/numba/cuda/models.py

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

numba_cuda/numba/cuda/nvvmutils.py

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

numba_cuda/numba/cuda/printimpl.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 singledispatch
 from llvmlite import ir
 from numba.core import types

numba_cuda/numba/cuda/random.py

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

numba_cuda/numba/cuda/reshape_funcs.cu

@@ -1,3 +1,8 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
 /*
  * Handle reshaping of zero-sized array.
  * See numba_attempt_nocopy_reshape() below.

numba_cuda/numba/cuda/serialize.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 Serialization support for compiled functions.
 """

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 *

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
 """

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.

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

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

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

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

@@ -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

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,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

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

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

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