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

numba_cuda/numba/cuda/tests/doc_examples/test_ffi.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 # Contents in this file are referenced from the sphinx-generated docs.
 # "magictoken" is used for markers as beginning and ending of example text.
 
@@ -87,7 +90,8 @@ class TestFFI(CUDATestCase):
 
     def test_ex_extra_includes(self):
         import numpy as np
-        from numba import cuda, config
+        from numba import cuda
+        from numba.cuda import config
         import os
 
         basedir = os.path.dirname(os.path.abspath(__file__))
@@ -100,7 +104,7 @@ class TestFFI(CUDATestCase):
         includedir = ":".join([mul_dir, add_dir])
         with override_config("CUDA_NVRTC_EXTRA_SEARCH_PATHS", includedir):
             # magictoken.ex_extra_search_paths.begin
-            from numba import config
+            from numba.cuda import config
 
             includedir = ":".join([mul_dir, add_dir])
             config.CUDA_NVRTC_EXTRA_SEARCH_PATHS = includedir

numba_cuda/numba/cuda/tests/doc_examples/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 unittest
 
 from numba.cuda.testing import (
@@ -5,14 +8,12 @@ from numba.cuda.testing import (
     skip_if_cudadevrt_missing,
     skip_on_cudasim,
     skip_unless_cc_60,
-    skip_if_mvc_enabled,
 )
 from numba.cuda.tests.support import captured_stdout
 
 
 @skip_if_cudadevrt_missing
 @skip_unless_cc_60
-@skip_if_mvc_enabled("CG not supported with MVC")
 @skip_on_cudasim("cudasim doesn't support cuda import at non-top-level")
 class TestLaplace(CUDATestCase):
     """

numba_cuda/numba/cuda/tests/doc_examples/test_matmul.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 """
 Matrix multiplication example via `cuda.jit`.
 

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

numba_cuda/numba/cuda/tests/doc_examples/test_random.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 # Contents in this file are referenced from the sphinx-generated docs.
 # "magictoken" is used for markers as beginning and ending of example text.
 

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

numba_cuda/numba/cuda/tests/doc_examples/test_sessionize.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 (
@@ -5,14 +8,12 @@ from numba.cuda.testing import (
     skip_if_cudadevrt_missing,
     skip_on_cudasim,
     skip_unless_cc_60,
-    skip_if_mvc_enabled,
 )
 from numba.cuda.tests.support import captured_stdout
 
 
 @skip_if_cudadevrt_missing
 @skip_unless_cc_60
-@skip_if_mvc_enabled("CG not supported with MVC")
 @skip_on_cudasim("cudasim doesn't support cuda import at non-top-level")
 class TestSessionization(CUDATestCase):
     """

numba_cuda/numba/cuda/tests/doc_examples/test_ufunc.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 CUDATestCase, skip_on_cudasim

numba_cuda/numba/cuda/tests/doc_examples/test_vecadd.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 CUDATestCase, skip_on_cudasim

numba_cuda/numba/cuda/tests/enum_usecases.py

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

numba_cuda/numba/cuda/tests/nocuda/__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/nocuda/test_dummyarray.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 itertools
 import numpy as np

numba_cuda/numba/cuda/tests/nocuda/test_function_resolution.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, skip_on_cudasim
 import operator
 from numba.core import types, typing

numba_cuda/numba/cuda/tests/nocuda/test_import.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.support import run_in_subprocess
 import unittest
 
@@ -23,6 +26,9 @@ class TestImport(unittest.TestCase):
             "numba.cpython.mathimpl",
             "numba.cpython.printimpl",
             "numba.cpython.randomimpl",
+            "numba.cuda.cpython.numbers",
+            "numba.cuda.cpython.cmathimpl",
+            "numba.cuda.cpython.mathimpl",
             "numba.core.optional",
             "numba.misc.gdb_hook",
             "numba.misc.literal",
@@ -34,7 +40,6 @@ class TestImport(unittest.TestCase):
             "numba.np.npyimpl",
             "numba.typed.typeddict",
             "numba.typed.typedlist",
-            "numba.experimental.jitclass.base",
         )
 
         code = "import sys; from numba import cuda; print(list(sys.modules))"

numba_cuda/numba/cuda/tests/nocuda/test_library_lookup.py

@@ -1,9 +1,13 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import sys
 import os
 import multiprocessing as mp
 import warnings
 
-from numba.core.config import IS_WIN32, IS_OSX
+
+from numba.cuda.core.config import IS_WIN32
 from numba.core.errors import NumbaWarning
 from numba.cuda.cudadrv import nvvm
 from numba.cuda.testing import (
@@ -16,6 +20,7 @@ from numba.cuda.cuda_paths import (
     _get_nvvm_path_decision,
     _get_cudalib_dir_path_decision,
     get_system_ctk,
+    get_system_ctk_libdir,
 )
 
 
@@ -99,10 +104,12 @@ class TestLibDeviceLookUp(LibraryLookupBase):
         # Check that CUDA_HOME works by removing conda-env
         by, info, warns = self.remote_do(self.do_set_cuda_home)
         self.assertEqual(by, "CUDA_HOME")
-        self.assertEqual(info, os.path.join("mycudahome", "nvvm", "libdevice"))
+        self.assertTrue(
+            info.startswith(os.path.join("mycudahome", "nvvm", "libdevice"))
+        )
         self.assertFalse(warns)
 
-        if get_system_ctk() is None:
+        if get_system_ctk("nvvm", "libdevice") is None:
             # Fake remove conda environment so no cudatoolkit is available
             by, info, warns = self.remote_do(self.do_clear_envs)
             self.assertEqual(by, "<unknown>")
@@ -145,13 +152,16 @@ class TestNvvmLookUp(LibraryLookupBase):
         self.assertEqual(by, "CUDA_HOME")
         self.assertFalse(warns)
         if IS_WIN32:
-            self.assertEqual(info, os.path.join("mycudahome", "nvvm", "bin"))
-        elif IS_OSX:
-            self.assertEqual(info, os.path.join("mycudahome", "nvvm", "lib"))
+            self.assertEqual(
+                os.path.dirname(info), os.path.join("mycudahome", "nvvm", "bin")
+            )
         else:
-            self.assertEqual(info, os.path.join("mycudahome", "nvvm", "lib64"))
+            self.assertEqual(
+                os.path.dirname(info),
+                os.path.join("mycudahome", "nvvm", "lib64"),
+            )
 
-        if get_system_ctk() is None:
+        if get_system_ctk("nvvm") is None:
             # Fake remove conda environment so no cudatoolkit is available
             by, info, warns = self.remote_do(self.do_clear_envs)
             self.assertEqual(by, "<unknown>")
@@ -196,12 +206,17 @@ class TestCudaLibLookUp(LibraryLookupBase):
         self.assertEqual(by, "CUDA_HOME")
         self.assertFalse(warns)
         if IS_WIN32:
-            self.assertEqual(info, os.path.join("mycudahome", "bin"))
-        elif IS_OSX:
-            self.assertEqual(info, os.path.join("mycudahome", "lib"))
+            # I think only wheels don't have the "Library" directory?
+            self.assertTrue(
+                info
+                in (
+                    os.path.join("mycudahome", "bin"),
+                    os.path.join("mycudahome", "Library", "bin"),
+                )
+            )
         else:
             self.assertEqual(info, os.path.join("mycudahome", "lib64"))
-        if get_system_ctk() is None:
+        if get_system_ctk_libdir() is None:
             # Fake remove conda environment so no cudatoolkit is available
             by, info, warns = self.remote_do(self.do_clear_envs)
             self.assertEqual(by, "<unknown>")

numba_cuda/numba/cuda/tests/nocuda/test_nvvm.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 nvvm
 from numba.cuda.testing import skip_on_cudasim
 from numba.cuda import utils

numba_cuda/numba/cuda/tests/nrt/__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/nrt/test_nrt.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 os
 
@@ -7,9 +10,10 @@ from numba.cuda.testing import CUDATestCase, skip_on_cudasim
 from numba.cuda.tests.support import run_in_subprocess, override_config
 from numba.cuda import get_current_device
 from numba.cuda.cudadrv.nvrtc import compile
-from numba import config, types
+from numba import types
 from numba.core.typing import signature
 from numba import cuda
+from numba.cuda import config
 from numba.cuda.typing.templates import AbstractTemplate
 from numba.cuda.cudadrv.linkable_code import (
     CUSource,

numba_cuda/numba/cuda/tests/nrt/test_nrt_refct.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 unittest
 from numba.cuda.tests.support import override_config

numba_cuda/numba/cuda/tests/support.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 import cmath
 import contextlib
 import enum
@@ -18,8 +21,9 @@ from functools import cached_property
 import numpy as np
 
 from numba import types
-from numba.core import errors, config
-from numba.core.typing import cffi_utils
+from numba.core import errors
+from numba.cuda.core import config
+from numba.cuda.typing import cffi_utils
 from numba.cuda.memory_management.nrt import rtsys
 from numba.core.extending import (
     typeof_impl,
@@ -28,7 +32,7 @@ from numba.core.extending import (
     NativeValue,
 )
 from numba.core.datamodel.models import OpaqueModel
-from numba.np import numpy_support
+from numba.cuda.np import numpy_support
 
 
 class EnableNRTStatsMixin(object):
@@ -748,16 +752,55 @@ class TestCase(unittest.TestCase):
 
         return Dummy, DummyType
 
-    def skip_if_no_external_compiler(self):
-        """
-        Call this to ensure the test is skipped if no suitable external compiler
-        is found. This is a method on the TestCase opposed to a stand-alone
-        decorator so as to make it "lazy" via runtime evaluation opposed to
-        running at test-discovery time.
-        """
-        # This is a local import to avoid deprecation warnings being generated
-        # through the use of the numba.pycc module.
-        from numba.pycc.platform import external_compiler_works
 
-        if not external_compiler_works():
-            self.skipTest("No suitable external compiler was found.")
+class MemoryLeak(object):
+    __enable_leak_check = True
+
+    def memory_leak_setup(self):
+        # Clean up any NRT-backed objects hanging in a dead reference cycle
+        gc.collect()
+        self.__init_stats = rtsys.get_allocation_stats()
+
+    def memory_leak_teardown(self):
+        if self.__enable_leak_check:
+            self.assert_no_memory_leak()
+
+    def assert_no_memory_leak(self):
+        old = self.__init_stats
+        new = rtsys.get_allocation_stats()
+        total_alloc = new.alloc - old.alloc
+        total_free = new.free - old.free
+        total_mi_alloc = new.mi_alloc - old.mi_alloc
+        total_mi_free = new.mi_free - old.mi_free
+        self.assertEqual(total_alloc, total_free)
+        self.assertEqual(total_mi_alloc, total_mi_free)
+
+    def disable_leak_check(self):
+        # For per-test use when MemoryLeakMixin is injected into a TestCase
+        self.__enable_leak_check = False
+
+
+class MemoryLeakMixin(EnableNRTStatsMixin, MemoryLeak):
+    def setUp(self):
+        super(MemoryLeakMixin, self).setUp()
+        self.memory_leak_setup()
+
+    def tearDown(self):
+        gc.collect()
+        self.memory_leak_teardown()
+        super(MemoryLeakMixin, self).tearDown()
+
+
+class CheckWarningsMixin(object):
+    @contextlib.contextmanager
+    def check_warnings(self, messages, category=RuntimeWarning):
+        with warnings.catch_warnings(record=True) as catch:
+            warnings.simplefilter("always")
+            yield
+        found = 0
+        for w in catch:
+            for m in messages:
+                if m in str(w.message):
+                    self.assertEqual(w.category, category)
+                    found += 1
+        self.assertEqual(found, len(messages))

numba_cuda/numba/cuda/tests/test_binary_generation/Makefile

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 # Generates the input files used by the nvjitlink tests
 
 # Test binaries are built taking into account the CC of the GPU in the test machine

numba_cuda/numba/cuda/tests/test_binary_generation/generate_raw_ltoir.py

@@ -1,4 +1,5 @@
-# Copyright (c) 2024, NVIDIA CORPORATION.
+# SPDX-FileCopyrightText: Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
 
 import argparse
 import os

numba_cuda/numba/cuda/tests/test_binary_generation/nrt_extern.cu

@@ -1,3 +1,8 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
 #include <nrt.cuh>
 
 extern "C" __device__ int device_allocate_deallocate(int* nb_retval){

numba_cuda/numba/cuda/tests/test_binary_generation/test_device_functions.cu

@@ -1,3 +1,8 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
 #include <cuda_fp16.h>
 
 extern __device__ bool __heq(__half arg1, __half arg2);

numba_cuda/numba/cuda/tests/test_binary_generation/undefined_extern.cu

@@ -1,3 +1,8 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
 extern __device__ float undef(float a, float b);
 
 __global__ void f(float *r, float *a, float *b) { r[0] = undef(a[0], b[0]); }

numba_cuda/numba/cuda/tests/test_tracing.py

@@ -0,0 +1,200 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
+from io import StringIO
+import logging
+
+import unittest
+from numba.cuda.core import tracing
+
+logger = logging.getLogger("trace")
+logger.setLevel(logging.INFO)
+
+# Make sure tracing is enabled
+orig_trace = tracing.trace
+tracing.trace = tracing.dotrace
+
+
+class CapturedTrace:
+    """Capture the trace temporarily for validation."""
+
+    def __init__(self):
+        self.buffer = StringIO()
+        self.handler = logging.StreamHandler(self.buffer)
+
+    def __enter__(self):
+        self._handlers = logger.handlers
+        self.buffer = StringIO()
+        logger.handlers = [logging.StreamHandler(self.buffer)]
+
+    def __exit__(self, type, value, traceback):
+        logger.handlers = self._handlers
+
+    def getvalue(self):
+        # Depending on how the tests are run, object names may be
+        # qualified by their containing module.
+        # Remove that to make the trace output independent from the testing mode.
+        log = self.buffer.getvalue()
+        log = log.replace(__name__ + ".", "")
+        return log
+
+
+class Class(object):
+    @tracing.trace
+    @classmethod
+    def class_method(cls):
+        pass
+
+    @tracing.trace
+    @staticmethod
+    def static_method():
+        pass
+
+    __test = None
+
+    def _test_get(self):
+        return self.__test
+
+    def _test_set(self, value):
+        self.__test = value
+
+    test = tracing.trace(property(_test_get, _test_set))
+
+    @tracing.trace
+    def method(self, some, other="value", *args, **kwds):
+        pass
+
+    def __repr__(self):
+        """Generate a deterministic string for testing."""
+        return "<Class instance>"
+
+
+class Class2(object):
+    @classmethod
+    def class_method(cls):
+        pass
+
+    @staticmethod
+    def static_method():
+        pass
+
+    __test = None
+
+    @property
+    def test(self):
+        return self.__test
+
+    @test.setter
+    def test(self, value):
+        self.__test = value
+
+    def method(self):
+        pass
+
+    def __str__(self):
+        return "Test(" + str(self.test) + ")"
+
+    def __repr__(self):
+        """Generate a deterministic string for testing."""
+        return "<Class2 instance>"
+
+
+@tracing.trace
+def test_traced_function():
+    # Test the tracing functionality with fixed values
+    x, y = 5, 5
+    z = True
+
+    a = x + y
+    b = x * y
+    if z:
+        result = a
+    else:
+        result = b
+
+    # The function should return 10 (5 + 5) when z is True
+    assert result == 10
+
+
+class TestTracing(unittest.TestCase):
+    def __init__(self, *args):
+        super(TestTracing, self).__init__(*args)
+
+    def setUp(self):
+        self.capture = CapturedTrace()
+
+    def tearDown(self):
+        del self.capture
+
+    def test_method(self):
+        with self.capture:
+            Class().method("foo", bar="baz")
+        self.assertEqual(
+            self.capture.getvalue(),
+            ">> Class.method(self=<Class instance>, some='foo', other='value', bar='baz')\n"
+            + "<< Class.method\n",
+        )
+
+    def test_class_method(self):
+        with self.capture:
+            Class.class_method()
+        self.assertEqual(
+            self.capture.getvalue(),
+            ">> Class.class_method(cls=<class 'Class'>)\n"
+            + "<< Class.class_method\n",
+        )
+
+    def test_static_method(self):
+        with self.capture:
+            Class.static_method()
+        self.assertEqual(
+            self.capture.getvalue(),
+            ">> static_method()\n" + "<< static_method\n",
+        )
+
+    def test_property(self):
+        with self.capture:
+            test = Class()
+            test.test = 1
+            assert 1 == test.test
+        self.assertEqual(
+            self.capture.getvalue(),
+            ">> Class._test_set(self=<Class instance>, value=1)\n"
+            + "<< Class._test_set\n"
+            + ">> Class._test_get(self=<Class instance>)\n"
+            + "<< Class._test_get -> 1\n",
+        )
+
+    def test_function(self):
+        with self.capture:
+            test_traced_function()
+        # The test function should be traced when called
+        trace_output = self.capture.getvalue()
+        self.assertIn(">> test_traced_function()", trace_output)
+        self.assertIn("<< test_traced_function", trace_output)
+
+    @unittest.skip("recursive decoration not yet implemented")
+    def test_injected(self):
+        with self.capture:
+            tracing.trace(Class2, recursive=True)
+            Class2.class_method()
+            Class2.static_method()
+            test = Class2()
+            test.test = 1
+            assert 1 == test.test
+            test.method()
+
+            self.assertEqual(
+                self.capture.getvalue(),
+                ">> Class2.class_method(cls=<type 'Class2'>)\n"
+                + "<< Class2.class_method\n"
+                ">> static_method()\n"
+                "<< static_method\n",
+            )
+
+
+# Reset tracing to its original value
+tracing.trace = orig_trace
+
+if __name__ == "__main__":
+    unittest.main()