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

numba_cuda/numba/cuda/tests/cudapy/test_ipc.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 itertools
 import traceback
@@ -14,7 +17,7 @@ from numba.cuda.testing import (
     ContextResettingTestCase,
     ForeignArray,
 )
-from numba.tests.support import linux_only, windows_only
+from numba.cuda.tests.support import linux_only, windows_only
 import unittest
 
 

numba_cuda/numba/cuda/tests/cudapy/test_ir_utils.py

@@ -0,0 +1,298 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
+from numba.cuda.testing import CUDATestCase
+import numba
+from numba.core.registry import cpu_target
+from numba.core.compiler import CompilerBase, Flags
+from numba.core.compiler_machinery import PassManager
+from numba.cuda.core import ir_utils
+from numba.core import types, ir, bytecode, compiler, registry
+from numba.core.untyped_passes import (
+    ExtractByteCode,
+    TranslateByteCode,
+    FixupArgs,
+    IRProcessing,
+)
+from numba.experimental import jitclass
+from numba.core.typed_passes import (
+    NopythonTypeInference,
+    type_inference_stage,
+    DeadCodeElimination,
+)
+
+
+# global constant for testing find_const
+GLOBAL_B = 11
+
+
+@jitclass([("val", numba.core.types.List(numba.intp))])
+class Dummy(object):
+    def __init__(self, val):
+        self.val = val
+
+
+class TestIrUtils(CUDATestCase):
+    """
+    Tests ir handling utility functions like find_callname.
+    """
+
+    def test_obj_func_match(self):
+        """Test matching of an object method (other than Array see #3449)"""
+
+        def test_func():
+            d = Dummy([1])
+            d.val.append(2)
+
+        test_ir = compiler.run_frontend(test_func)
+        typingctx = cpu_target.typing_context
+        targetctx = cpu_target.target_context
+        typing_res = type_inference_stage(
+            typingctx, targetctx, test_ir, (), None
+        )
+        matched_call = ir_utils.find_callname(
+            test_ir, test_ir.blocks[0].body[7].value, typing_res.typemap
+        )
+        self.assertTrue(
+            isinstance(matched_call, tuple)
+            and len(matched_call) == 2
+            and matched_call[0] == "append"
+        )
+
+    def test_dead_code_elimination(self):
+        class Tester(CompilerBase):
+            @classmethod
+            def mk_pipeline(
+                cls,
+                args,
+                return_type=None,
+                flags=None,
+                locals=None,
+                library=None,
+                typing_context=None,
+                target_context=None,
+            ):
+                if locals is None:
+                    locals = {}
+                if not flags:
+                    flags = Flags()
+                flags.nrt = True
+                if typing_context is None:
+                    typing_context = registry.cpu_target.typing_context
+                if target_context is None:
+                    target_context = registry.cpu_target.target_context
+                return cls(
+                    typing_context,
+                    target_context,
+                    library,
+                    args,
+                    return_type,
+                    flags,
+                    locals,
+                )
+
+            def compile_to_ir(self, func, DCE=False):
+                """
+                Compile and return IR
+                """
+                func_id = bytecode.FunctionIdentity.from_function(func)
+                self.state.func_id = func_id
+                ExtractByteCode().run_pass(self.state)
+                state = self.state
+
+                name = "DCE_testing"
+                pm = PassManager(name)
+                pm.add_pass(TranslateByteCode, "analyzing bytecode")
+                pm.add_pass(FixupArgs, "fix up args")
+                pm.add_pass(IRProcessing, "processing IR")
+                pm.add_pass(NopythonTypeInference, "nopython frontend")
+                if DCE is True:
+                    pm.add_pass(DeadCodeElimination, "DCE after typing")
+                pm.finalize()
+                pm.run(state)
+                return state.func_ir
+
+        def check_initial_ir(the_ir):
+            # dead stuff:
+            # a const int value 0xdead
+            # an assign of above into to variable `dead`
+            # a const int above 0xdeaddead
+            # an assign of said int to variable `deaddead`
+            # this is 2 statements to remove
+
+            self.assertEqual(len(the_ir.blocks), 1)
+            block = the_ir.blocks[0]
+            deads = []
+            for x in block.find_insts(ir.Assign):
+                if isinstance(getattr(x, "target", None), ir.Var):
+                    if "dead" in getattr(x.target, "name", ""):
+                        deads.append(x)
+
+            self.assertEqual(len(deads), 2)
+            for d in deads:
+                # check the ir.Const is the definition and the value is expected
+                const_val = the_ir.get_definition(d.value)
+                self.assertTrue(
+                    int("0x%s" % d.target.name, 16), const_val.value
+                )
+
+            return deads
+
+        def check_dce_ir(the_ir):
+            self.assertEqual(len(the_ir.blocks), 1)
+            block = the_ir.blocks[0]
+            deads = []
+            consts = []
+            for x in block.find_insts(ir.Assign):
+                if isinstance(getattr(x, "target", None), ir.Var):
+                    if "dead" in getattr(x.target, "name", ""):
+                        deads.append(x)
+                if isinstance(getattr(x, "value", None), ir.Const):
+                    consts.append(x)
+            self.assertEqual(len(deads), 0)
+
+            # check the consts to make sure there's no reference to 0xdead or
+            # 0xdeaddead
+            for x in consts:
+                self.assertTrue(x.value.value not in [0xDEAD, 0xDEADDEAD])
+
+        def foo(x):
+            y = x + 1
+            dead = 0xDEAD  # noqa
+            z = y + 2
+            deaddead = 0xDEADDEAD  # noqa
+            ret = z * z
+            return ret
+
+        test_pipeline = Tester.mk_pipeline((types.intp,))
+        no_dce = test_pipeline.compile_to_ir(foo)
+        removed = check_initial_ir(no_dce)
+
+        test_pipeline = Tester.mk_pipeline((types.intp,))
+        w_dce = test_pipeline.compile_to_ir(foo, DCE=True)
+        check_dce_ir(w_dce)
+
+        # check that the count of initial - removed = dce
+        self.assertEqual(
+            len(no_dce.blocks[0].body) - len(removed), len(w_dce.blocks[0].body)
+        )
+
+    def test_find_const_global(self):
+        """
+        Test find_const() for values in globals (ir.Global) and freevars
+        (ir.FreeVar) that are considered constants for compilation.
+        """
+        FREEVAR_C = 12
+
+        def foo(a):
+            b = GLOBAL_B
+            c = FREEVAR_C
+            return a + b + c
+
+        f_ir = compiler.run_frontend(foo)
+        block = f_ir.blocks[0]
+        const_b = None
+        const_c = None
+
+        for inst in block.body:
+            if isinstance(inst, ir.Assign) and inst.target.name == "b":
+                const_b = ir_utils.guard(ir_utils.find_const, f_ir, inst.target)
+            if isinstance(inst, ir.Assign) and inst.target.name == "c":
+                const_c = ir_utils.guard(ir_utils.find_const, f_ir, inst.target)
+
+        self.assertEqual(const_b, GLOBAL_B)
+        self.assertEqual(const_c, FREEVAR_C)
+
+    def test_flatten_labels(self):
+        """tests flatten_labels"""
+
+        def foo(a):
+            acc = 0
+            if a > 3:
+                acc += 1
+                if a > 19:
+                    return 53
+            elif a < 1000:
+                if a >= 12:
+                    acc += 1
+                for x in range(10):
+                    acc -= 1
+                    if acc < 2:
+                        break
+                else:
+                    acc += 7
+            else:
+                raise ValueError("some string")
+            # prevents inline of return on py310
+            py310_defeat1 = 1  # noqa
+            py310_defeat2 = 2  # noqa
+            py310_defeat3 = 3  # noqa
+            py310_defeat4 = 4  # noqa
+            return acc
+
+        def bar(a):
+            acc = 0
+            z = 12
+            if a > 3:
+                acc += 1
+                z += 12
+                if a > 19:
+                    z += 12
+                    return 53
+            elif a < 1000:
+                if a >= 12:
+                    z += 12
+                    acc += 1
+                for x in range(10):
+                    z += 12
+                    acc -= 1
+                    if acc < 2:
+                        break
+                else:
+                    z += 12
+                    acc += 7
+            else:
+                raise ValueError("some string")
+            py310_defeat1 = 1  # noqa
+            py310_defeat2 = 2  # noqa
+            py310_defeat3 = 3  # noqa
+            py310_defeat4 = 4  # noqa
+            return acc
+
+        def baz(a):
+            acc = 0
+            if a > 3:
+                acc += 1
+                if a > 19:
+                    return 53
+                else:  # extra control flow in comparison to foo
+                    return 55
+            elif a < 1000:
+                if a >= 12:
+                    acc += 1
+                for x in range(10):
+                    acc -= 1
+                    if acc < 2:
+                        break
+                else:
+                    acc += 7
+            else:
+                raise ValueError("some string")
+            py310_defeat1 = 1  # noqa
+            py310_defeat2 = 2  # noqa
+            py310_defeat3 = 3  # noqa
+            py310_defeat4 = 4  # noqa
+            return acc
+
+        def get_flat_cfg(func):
+            func_ir = ir_utils.compile_to_numba_ir(func, dict())
+            flat_blocks = ir_utils.flatten_labels(func_ir.blocks)
+            self.assertEqual(max(flat_blocks.keys()) + 1, len(func_ir.blocks))
+            return ir_utils.compute_cfg_from_blocks(flat_blocks)
+
+        foo_cfg = get_flat_cfg(foo)
+        bar_cfg = get_flat_cfg(bar)
+        baz_cfg = get_flat_cfg(baz)
+
+        self.assertEqual(foo_cfg, bar_cfg)
+        self.assertNotEqual(foo_cfg, baz_cfg)

numba_cuda/numba/cuda/tests/cudapy/test_iterators.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 unittest, CUDATestCase
 

numba_cuda/numba/cuda/tests/cudapy/test_lang.py

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

numba_cuda/numba/cuda/tests/cudapy/test_laplace.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, float64, void
 from numba.cuda.testing import unittest, CUDATestCase

numba_cuda/numba/cuda/tests/cudapy/test_libdevice.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.core import types
 from numba.cuda.testing import skip_on_cudasim, unittest, CUDATestCase

numba_cuda/numba/cuda/tests/cudapy/test_lineinfo.py

@@ -1,7 +1,10 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from numba import cuda, float32, int32
 from numba.core.errors import NumbaInvalidConfigWarning
 from numba.cuda.testing import CUDATestCase, skip_on_cudasim
-from numba.tests.support import ignore_internal_warnings
+from numba.cuda.tests.support import ignore_internal_warnings
 import re
 import unittest
 import warnings

numba_cuda/numba/cuda/tests/cudapy/test_localmem.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, int32, complex128, void

numba_cuda/numba/cuda/tests/cudapy/test_mandel.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 float64, uint32
 from numba.cuda.compiler import compile_ptx
 from numba.cuda.testing import skip_on_cudasim, unittest

numba_cuda/numba/cuda/tests/cudapy/test_math.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.testing import (
     skip_unless_cc_53,

numba_cuda/numba/cuda/tests/cudapy/test_matmul.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, float32, void

numba_cuda/numba/cuda/tests/cudapy/test_minmax.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, float64

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

numba_cuda/numba/cuda/tests/cudapy/test_multigpu.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
 import numpy as np
 from numba.cuda.testing import skip_on_cudasim, CUDATestCase

numba_cuda/numba/cuda/tests/cudapy/test_multiprocessing.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 multiprocessing as mp
 

numba_cuda/numba/cuda/tests/cudapy/test_multithreads.py

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

numba_cuda/numba/cuda/tests/cudapy/test_nondet.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, float32, void
 from numba.cuda.testing import unittest, CUDATestCase

numba_cuda/numba/cuda/tests/cudapy/test_operator.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.testing import (
     unittest,
@@ -9,7 +12,7 @@ from numba.cuda.testing import (
 from numba import cuda
 from numba.core import types
 from numba.core.types import f2, b1
-from numba.core.typing import signature
+from numba.cuda.typing import signature
 import operator
 import itertools
 from numba.np.numpy_support import from_dtype

numba_cuda/numba/cuda/tests/cudapy/test_optimization.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.testing import skip_on_cudasim, CUDATestCase

numba_cuda/numba/cuda/tests/cudapy/test_overload.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, njit, types, version_info
 from numba.core.errors import TypingError
 from numba.core.extending import overload, overload_attribute

numba_cuda/numba/cuda/tests/cudapy/test_powi.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 numpy as np
 from numba import cuda, float64, int8, int32, void

numba_cuda/numba/cuda/tests/cudapy/test_print.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 CUDATestCase, skip_on_cudasim
 import numpy as np
 import subprocess

numba_cuda/numba/cuda/tests/cudapy/test_py2_div_issue.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, float32, int32, void
 from numba.cuda.testing import unittest, CUDATestCase

numba_cuda/numba/cuda/tests/cudapy/test_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
 
 import numpy as np

numba_cuda/numba/cuda/tests/cudapy/test_record_dtype.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.core import types

numba_cuda/numba/cuda/tests/cudapy/test_recursion.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.core.errors import TypingError
 from numba.cuda.testing import CUDATestCase, skip_on_cudasim

numba_cuda/numba/cuda/tests/cudapy/test_reduction.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.core.config import ENABLE_CUDASIM

numba_cuda/numba/cuda/tests/cudapy/test_retrieve_autoconverted_arrays.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

numba_cuda/numba/cuda/tests/cudapy/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 pickle
 import numpy as np
 from numba import cuda, vectorize

numba_cuda/numba/cuda/tests/cudapy/test_slicing.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/cudapy/test_sm.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, int32, float64, void
 from numba.core.errors import TypingError
 from numba.core import types

numba_cuda/numba/cuda/tests/cudapy/test_sm_creation.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, float32, int32, void
 from numba.core.errors import TypingError

numba_cuda/numba/cuda/tests/cudapy/test_stream_api.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 (
     skip_on_cudasim,
     skip_unless_cudasim,

numba_cuda/numba/cuda/tests/cudapy/test_sync.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, int32, float32
 from numba.cuda.testing import skip_on_cudasim, unittest, CUDATestCase

numba_cuda/numba/cuda/tests/cudapy/test_transpose.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.kernels.transpose import transpose

numba_cuda/numba/cuda/tests/cudapy/test_ufuncs.py

@@ -1,9 +1,12 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import functools
 import numpy as np
 import unittest
 
 from numba import config, cuda, types
-from numba.tests.support import TestCase
+from numba.cuda.tests.support import TestCase
 from numba.tests.test_ufuncs import BasicUFuncTest
 
 

numba_cuda/numba/cuda/tests/cudapy/test_userexc.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, CUDATestCase
 from numba import cuda
 from numba.core import config

numba_cuda/numba/cuda/tests/cudapy/test_vector_type.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 CUDA vector type tests. Note that this test file imports
 `cuda.vector_type` module to programmatically test all the

numba_cuda/numba/cuda/tests/cudapy/test_vectorize.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/tests/cudapy/test_vectorize_complex.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 vectorize
 from numba.cuda.testing import skip_on_cudasim, CUDATestCase

numba_cuda/numba/cuda/tests/cudapy/test_vectorize_decor.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 vectorize, cuda

numba_cuda/numba/cuda/tests/cudapy/test_vectorize_device.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 vectorize
 from numba import cuda, float32
 import numpy as np

numba_cuda/numba/cuda/tests/cudapy/test_vectorize_scalar_arg.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 vectorize
 from numba import cuda, float64