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

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
@@ -57,8 +60,8 @@ class TestCUDAFindLibs(CUDATestCase):
     @unittest.skipIf(not sys.platform.startswith("linux"), "linux only")
     def test_cuda_find_lib_errors(self):
         """
-        This tests that the find_libs works as expected in the case of an
-        environment variable being used to set the path.
+        This tests that driver discovery attempts to load from typical system
+        locations and fails gracefully if pointed at an invalid directory.
         """
         # one of these is likely to exist on linux, it's also unlikely that
         # someone has extracted the contents of libdevice into here!
@@ -73,8 +76,11 @@ class TestCUDAFindLibs(CUDATestCase):
         # This is the testing part, the test will only run if there's a valid
         # path in which to look
         if looking_for is not None:
+            # We no longer support NUMBA_CUDA_DRIVER. Still run the subprocess
+            # to ensure importing and a trivial kernel launch path work, but
+            # do not set any Numba-specific driver env vars.
             out, err = self.run_test_in_separate_process(
-                "NUMBA_CUDA_DRIVER", looking_for
+                "DUMMY_UNUSED", looking_for
             )
             self.assertTrue(out is not None)
             self.assertTrue(err is not None)

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

@@ -1,9 +1,12 @@
+# 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
 
 from numba import cuda
-from numba.core import config
+from numba.cuda.core import config
 from numba.cuda.testing import unittest, CUDATestCase, skip_on_cudasim
 from numba.cuda.tests.support import linux_only
 

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,8 +1,15 @@
+# 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
+from numba.cuda import config
 from numba.cuda.testing import unittest
-from numba.cuda.testing import skip_on_cudasim, skip_if_cuda_includes_missing
+from numba.cuda.testing import (
+    skip_on_cudasim,
+    skip_if_cuda_includes_missing,
+    skip_if_nvjitlink_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
@@ -326,6 +333,18 @@ class TestLinker(CUDATestCase):
         calc_size = np.dtype(np.float64).itemsize * LMEM_SIZE
         self.assertGreaterEqual(local_mem_size, calc_size)
 
+    @skip_if_nvjitlink_missing("nvJitLink not installed or new enough (>12.3)")
+    def test_link_for_different_cc(self):
+        linker = _Linker.new(cc=(7, 5), lto=True)
+        code = """
+__device__ int foo(int x) {
+    return x + 1;
+}
+"""
+        linker.add_cu(code, "foo")
+        ptx = linker.get_linked_ptx().decode()
+        assert "target sm_75" in ptx
+
 
 if __name__ == "__main__":
     unittest.main()

numba_cuda/numba/cuda/tests/cudadrv/test_managed_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 ctypes import byref, c_size_t
 from numba.cuda.cudadrv.driver import device_memset, driver, USE_NV_BINDING

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

@@ -1,9 +1,13 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import unittest
 import threading
 
 import numpy as np
 
-from numba import cuda, config
+from numba import cuda
+from numba.cuda import config
 from numba.cuda.cudadrv.linkable_code import CUSource
 from numba.cuda.testing import (
     CUDATestCase,

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
@@ -5,7 +8,7 @@ from numba.cuda import get_current_device
 from numba.cuda.cudadrv.driver import _Linker, _have_nvjitlink
 
 from numba import cuda
-from numba import config
+from numba.cuda import config
 
 import os
 import io

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
 

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

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

numba_cuda/numba/cuda/tests/cudadrv/test_pinned.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 platform
 

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

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

numba_cuda/numba/cuda/tests/cudadrv/test_ptds.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 logging
 import traceback
@@ -12,7 +15,7 @@ from numba.cuda.tests.support import linux_only
 
 def child_test():
     from numba import cuda, int32, void
-    from numba.core import config
+    from numba.cuda.core import config
     import io
     import numpy as np
     import threading

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

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

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

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

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

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 #
 # Test does not work on some cards.
 #

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

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

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

numba_cuda/numba/cuda/tests/cudapy/cache_with_cpu_usecases.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 numba import cuda, njit

numba_cuda/numba/cuda/tests/cudapy/cg_cache_usecases.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
 import sys

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

@@ -1,5 +1,8 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from numba import types
-from numba.core import config
+from numba.cuda.core import config
 
 
 class TestStruct:

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

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 Usecases of recursive functions in the CUDA target, many derived from
 numba/tests/recursion_usecases.py.

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

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

@@ -1,8 +1,12 @@
+# 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, CUDATestCase
 from numba.cuda.testing import skip_on_cudasim, skip_unless_cudasim
-from numba import config, cuda
+from numba import cuda
+from numba.cuda import config
 
 
 if config.ENABLE_CUDASIM:

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

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

numba_cuda/numba/cuda/tests/cudapy/test_array_args.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_array_methods.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 CUDATestCase

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

@@ -1,9 +1,12 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import numpy as np
 from textwrap import dedent
 
 from numba import cuda, uint32, uint64, float32, float64
 from numba.cuda.testing import unittest, CUDATestCase, cc_X_or_above
-from numba.core import config
+from numba.cuda.core import config
 
 
 @cuda.jit(device=True)