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

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
@@ -25,10 +28,6 @@ class TestNvvmDriver(unittest.TestCase):
         # ("-gen-lto") - all other NVVM options are of the form
         # "-<name>=<value>"
 
-        # -gen-lto is not available prior to CUDA 11.5
-        if runtime.get_version() < (11, 5):
-            self.skipTest("-gen-lto unavailable in this toolkit version")
-
         nvvmir = self.get_nvvmir()
         arch = "compute_%d%d" % nvrtc.get_lowest_supported_cc()
         ltoir = nvvm.compile_ir(nvvmir, opt=3, gen_lto=None, arch=arch)

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
@@ -7,7 +10,7 @@ from numba.cuda.testing import (
     skip_with_cuda_python,
     skip_under_cuda_memcheck,
 )
-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_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,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.core import config
 

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

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,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 textwrap import dedent
 

numba_cuda/numba/cuda/tests/cudapy/test_bfloat16.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, float32
 from numba.cuda.bf16 import bfloat16
 from numba.cuda.testing import CUDATestCase
@@ -8,9 +11,7 @@ import math
 class TestBfloat16HighLevelBindings(CUDATestCase):
     def skip_unsupported(self):
         if not cuda.is_bfloat16_supported():
-            self.skipTest(
-                "bfloat16 requires compute capability 8.0+ and CUDA version>= 12.0"
-            )
+            self.skipTest("bfloat16 requires compute capability 8.0+")
 
     def test_use_type_in_kernel(self):
         self.skip_unsupported()

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

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import numba.cuda as cuda
 from numba.cuda.testing import unittest, CUDATestCase
 import numpy as np
@@ -43,9 +46,7 @@ dtypes = [int16, int32, int64, uint16, uint32, uint64, float32]
 class Bfloat16Test(CUDATestCase):
     def skip_unsupported(self):
         if not cuda.is_bfloat16_supported():
-            self.skipTest(
-                "bfloat16 requires compute capability 8.0+ and CUDA version>= 12.0"
-            )
+            self.skipTest("bfloat16 requires compute capability 8.0+")
 
     def test_ctor(self):
         self.skip_unsupported()

numba_cuda/numba/cuda/tests/cudapy/test_blackscholes.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 math
 from numba import cuda, double, void

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

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

@@ -1,8 +1,14 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import multiprocessing
 import os
 import shutil
 import unittest
 import warnings
+import sys
+import stat
+import subprocess
 
 from numba import cuda
 from numba.core.errors import NumbaWarning
@@ -14,12 +20,152 @@ from numba.cuda.testing import (
     skip_if_mvc_enabled,
     test_data_dir,
 )
-from numba.tests.test_caching import (
-    DispatcherCacheUsecasesTest,
-    skip_bad_access,
+from numba.cuda.tests.support import (
+    TestCase,
+    temp_directory,
+    import_dynamic,
 )
 
 
+class BaseCacheTest(TestCase):
+    # The source file that will be copied
+    usecases_file = None
+    # Make sure this doesn't conflict with another module
+    modname = None
+
+    def setUp(self):
+        self.tempdir = temp_directory("test_cache")
+        sys.path.insert(0, self.tempdir)
+        self.modfile = os.path.join(self.tempdir, self.modname + ".py")
+        self.cache_dir = os.path.join(self.tempdir, "__pycache__")
+        shutil.copy(self.usecases_file, self.modfile)
+        os.chmod(self.modfile, stat.S_IREAD | stat.S_IWRITE)
+        self.maxDiff = None
+
+    def tearDown(self):
+        sys.modules.pop(self.modname, None)
+        sys.path.remove(self.tempdir)
+
+    def import_module(self):
+        # Import a fresh version of the test module.  All jitted functions
+        # in the test module will start anew and load overloads from
+        # the on-disk cache if possible.
+        old = sys.modules.pop(self.modname, None)
+        if old is not None:
+            # Make sure cached bytecode is removed
+            cached = [old.__cached__]
+            for fn in cached:
+                try:
+                    os.unlink(fn)
+                except FileNotFoundError:
+                    pass
+        mod = import_dynamic(self.modname)
+        self.assertEqual(mod.__file__.rstrip("co"), self.modfile)
+        return mod
+
+    def cache_contents(self):
+        try:
+            return [
+                fn
+                for fn in os.listdir(self.cache_dir)
+                if not fn.endswith((".pyc", ".pyo"))
+            ]
+        except FileNotFoundError:
+            return []
+
+    def get_cache_mtimes(self):
+        return dict(
+            (fn, os.path.getmtime(os.path.join(self.cache_dir, fn)))
+            for fn in sorted(self.cache_contents())
+        )
+
+    def check_pycache(self, n):
+        c = self.cache_contents()
+        self.assertEqual(len(c), n, c)
+
+    def dummy_test(self):
+        pass
+
+
+class DispatcherCacheUsecasesTest(BaseCacheTest):
+    here = os.path.dirname(__file__)
+    usecases_file = os.path.join(here, "cache_usecases.py")
+    modname = "dispatcher_caching_test_fodder"
+
+    def run_in_separate_process(self, *, envvars={}):
+        # Cached functions can be run from a distinct process.
+        # Also stresses issue #1603: uncached function calling cached function
+        # shouldn't fail compiling.
+        code = """if 1:
+            import sys
+
+            sys.path.insert(0, %(tempdir)r)
+            mod = __import__(%(modname)r)
+            mod.self_test()
+            """ % dict(tempdir=self.tempdir, modname=self.modname)
+
+        subp_env = os.environ.copy()
+        subp_env.update(envvars)
+        popen = subprocess.Popen(
+            [sys.executable, "-c", code],
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            env=subp_env,
+        )
+        out, err = popen.communicate()
+        if popen.returncode != 0:
+            raise AssertionError(
+                "process failed with code %s: \n"
+                "stdout follows\n%s\n"
+                "stderr follows\n%s\n"
+                % (popen.returncode, out.decode(), err.decode()),
+            )
+
+    def check_hits(self, func, hits, misses=None):
+        st = func.stats
+        self.assertEqual(sum(st.cache_hits.values()), hits, st.cache_hits)
+        if misses is not None:
+            self.assertEqual(
+                sum(st.cache_misses.values()), misses, st.cache_misses
+            )
+
+
+def check_access_is_preventable():
+    # This exists to check whether it is possible to prevent access to
+    # a file/directory through the use of `chmod 500`. If a user has
+    # elevated rights (e.g. root) then writes are likely to be possible
+    # anyway. Tests that require functioning access prevention are
+    # therefore skipped based on the result of this check.
+    tempdir = temp_directory("test_cache")
+    test_dir = os.path.join(tempdir, "writable_test")
+    os.mkdir(test_dir)
+    # check a write is possible
+    with open(os.path.join(test_dir, "write_ok"), "wt") as f:
+        f.write("check1")
+    # now forbid access
+    os.chmod(test_dir, 0o500)
+    try:
+        with open(os.path.join(test_dir, "write_forbidden"), "wt") as f:
+            f.write("check2")
+        # access prevention is not possible
+        return False
+    except PermissionError:
+        # Check that the cause of the exception is due to access/permission
+        # as per
+        # https://github.com/conda/conda/blob/4.5.0/conda/gateways/disk/permissions.py#L35-L37  # noqa: E501
+        # errno reports access/perm fail so access prevention via
+        # `chmod 500` works for this user.
+        return True
+    finally:
+        os.chmod(test_dir, 0o775)
+        shutil.rmtree(test_dir)
+
+
+_access_preventable = check_access_is_preventable()
+_access_msg = "Cannot create a directory to which writes are preventable"
+skip_bad_access = unittest.skipUnless(_access_preventable, _access_msg)
+
+
 @skip_on_cudasim("Simulator does not implement caching")
 class CUDACachingTest(DispatcherCacheUsecasesTest):
     here = os.path.dirname(__file__)

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

numba_cuda/numba/cuda/tests/cudapy/test_cffi.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, types
@@ -7,7 +10,7 @@ from numba.cuda.testing import (
     unittest,
     CUDATestCase,
 )
-from numba.tests.support import skip_unless_cffi
+from numba.cuda.tests.support import skip_unless_cffi
 
 
 @skip_unless_cffi

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

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from math import sqrt
 from numba import cuda, float32, int16, int32, int64, types, uint32, void
 from numba.cuda import (
@@ -6,7 +9,6 @@ from numba.cuda import (
     compile_ptx,
     compile_ptx_for_current_device,
 )
-from numba.cuda.cudadrv import runtime
 from numba.cuda.testing import skip_on_cudasim, unittest, CUDATestCase
 
 
@@ -226,9 +228,6 @@ class TestCompile(unittest.TestCase):
         )
 
     def test_compile_to_ltoir(self):
-        if runtime.get_version() < (11, 5):
-            self.skipTest("-gen-lto unavailable in this toolkit version")
-
         ltoir, resty = compile(
             f_module, int32(int32, int32), device=True, output="ltoir"
         )

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

numba_cuda/numba/cuda/tests/cudapy/test_complex_kernel.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_const_string.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 numpy as np
 from numba import cuda

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

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

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

numba_cuda/numba/cuda/tests/cudapy/test_cuda_array_interface.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 numba import vectorize, guvectorize
 from numba import cuda
 from numba.cuda.testing import unittest, ContextResettingTestCase, ForeignArray
 from numba.cuda.testing import skip_on_cudasim, skip_if_external_memmgr
-from numba.tests.support import linux_only, override_config
+from numba.cuda.tests.support import linux_only, override_config
 from unittest.mock import call, patch
 
 

numba_cuda/numba/cuda/tests/cudapy/test_cuda_jit_no_types.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
 import numpy as np
 from numba.cuda.testing import CUDATestCase
-from numba.tests.support import override_config
+from numba.cuda.tests.support import override_config
 import unittest
 
 

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

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

@@ -1,8 +1,11 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import numpy as np
 
 from numba.core.utils import PYVERSION
 from numba.cuda.testing import skip_on_cudasim, CUDATestCase
-from numba.tests.support import (
+from numba.cuda.tests.support import (
     override_config,
     captured_stderr,
     captured_stdout,