numba-cuda 0.18.0__tar.gz → 0.18.1__tar.gz

{numba_cuda-0.18.0 → numba_cuda-0.18.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: numba-cuda
-Version: 0.18.0
+Version: 0.18.1
 Summary: CUDA target for Numba
 Author: Anaconda Inc., NVIDIA Corporation
 License: BSD 2-clause

numba_cuda-0.18.1/numba_cuda/VERSION

@@ -0,0 +1 @@
+0.18.1

{numba_cuda-0.18.0 → numba_cuda-0.18.1}/numba_cuda/numba/cuda/dispatcher.py

@@ -772,6 +772,7 @@ class _FunctionCompiler(object):
         self.py_func = py_func
         self.targetdescr = targetdescr
         self.targetoptions = targetoptions
+        self.locals = {}
         self.pysig = utils.pysignature(self.py_func)
         self.pipeline_class = pipeline_class
         # Remember key=(args, return_type) combinations that will fail
@@ -842,7 +843,7 @@ class _FunctionCompiler(object):
             args=args,
             return_type=return_type,
             flags=flags,
-            locals={},
+            locals=self.locals,
             pipeline_class=self.pipeline_class,
         )
         # Check typing error if object mode is used

{numba_cuda-0.18.0 → numba_cuda-0.18.1}/numba_cuda/numba/cuda/tests/cudadrv/test_context_stack.py

@@ -128,11 +128,17 @@ class Test3rdPartyContext(CUDATestCase):
                 "Error getting CUDA driver version",
             )
 
-            # CUDA 13's cuCtxCreate has an optional parameter prepended
-            if version >= 13000:
-                args = (None, flags, dev)
-            else:
+            # CUDA 13's cuCtxCreate has an optional parameter prepended. The
+            # version of cuCtxCreate in use depends on the cuda.bindings major
+            # version rather than the installed driver version on the machine
+            # we're running on.
+            from cuda import bindings
+
+            bindings_version = int(bindings.__version__.split(".")[0])
+            if bindings_version in (11, 12):
                 args = (flags, dev)
+            else:
+                args = (None, flags, dev)
 
             hctx = the_driver.cuCtxCreate(*args)
         else:

{numba_cuda-0.18.0 → numba_cuda-0.18.1}/numba_cuda/numba/cuda/tests/cudapy/test_compiler.py

@@ -252,6 +252,21 @@ class TestCompile(unittest.TestCase):
                 output=illegal_output,
             )
 
+    def test_functioncompiler_locals(self):
+        # Tests against regression fixed in:
+        # https://github.com/NVIDIA/numba-cuda/pull/381
+        #
+        # "AttributeError: '_FunctionCompiler' object has no attribute
+        # 'locals'"
+        cond = None
+
+        @cuda.jit("void(float32[::1])")
+        def f(b_arg):
+            b_smem = cuda.shared.array(shape=(1,), dtype=float32)
+
+            if cond:
+                b_smem[0] = b_arg[0]
+
 
 @skip_on_cudasim("Compilation unsupported in the simulator")
 class TestCompileForCurrentDevice(CUDATestCase):

{numba_cuda-0.18.0 → numba_cuda-0.18.1}/numba_cuda/numba/cuda/tests/cudapy/test_inspect.py

@@ -178,8 +178,18 @@ class TestInspect(CUDATestCase):
 
         np.testing.assert_equal(arr[0], np.float16(1) + np.float16(2))
 
+    def skip_on_cuda_version_issues(self):
+        # FIXME: This should be unskipped once the cause of certain nvdisasm
+        # versions failing to dump SASS with certain driver / nvJitLink
+        # versions is understood
+        self.skipTest(
+            "Relocation information required for analysis not preserved"
+        )
+
     @skip_without_nvdisasm("nvdisasm needed for inspect_sass()")
     def test_inspect_sass_eager(self):
+        self.skip_on_cuda_version_issues()
+
         sig = (float32[::1], int32[::1])
 
         @cuda.jit(sig, lineinfo=True)
@@ -192,6 +202,8 @@ class TestInspect(CUDATestCase):
 
     @skip_without_nvdisasm("nvdisasm needed for inspect_sass()")
     def test_inspect_sass_lazy(self):
+        self.skip_on_cuda_version_issues()
+
         @cuda.jit(lineinfo=True)
         def add(x, y):
             i = cuda.grid(1)
@@ -220,6 +232,8 @@ class TestInspect(CUDATestCase):
 
     @skip_without_nvdisasm("nvdisasm needed for inspect_sass_cfg()")
     def test_inspect_sass_cfg(self):
+        self.skip_on_cuda_version_issues()
+
         sig = (float32[::1], int32[::1])
 
         @cuda.jit(sig)

{numba_cuda-0.18.0 → numba_cuda-0.18.1}/numba_cuda.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: numba-cuda
-Version: 0.18.0
+Version: 0.18.1
 Summary: CUDA target for Numba
 Author: Anaconda Inc., NVIDIA Corporation
 License: BSD 2-clause

numba_cuda-0.18.0/numba_cuda/VERSION

@@ -1 +0,0 @@
-0.18.0