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

numba_cuda/VERSION

@@ -1 +1 @@
-0.18.0
+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/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/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/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.dist-info → numba_cuda-0.18.1.dist-info}/METADATA

@@ -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.dist-info → numba_cuda-0.18.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 _numba_cuda_redirector.pth,sha256=cmfMMmV0JPh3yEpl4bGeM9AuXiVVMSo6Z_b7RaQL3XE,30
 _numba_cuda_redirector.py,sha256=n_r8MYbu5-vcXMnLJW147k8DnFXXvgb7nPIXnlXwTyQ,2659
-numba_cuda/VERSION,sha256=9ek4KWfJgzBmpVgIgp83UPyOEYGxcMr9Iiw3BIu0mnc,7
+numba_cuda/VERSION,sha256=jE-M5QqxQhZgmJCF5E6AErG5sq3jYNneuEXLqUczzVY,7
 numba_cuda/__init__.py,sha256=atXeUvJKR3JHcAiCFbXCVOJQUHgB1TulmsqSL_9RT3Q,114
 numba_cuda/_version.py,sha256=nzrrJXi85d18m6SPdsPsetJNClDETkmF1MrEhGLYDBs,734
 numba_cuda/numba/cuda/__init__.py,sha256=6MIFgqrhLldYgI11ZzVC4ANaW7Bs6c5Z5e7tqlEOemo,3353
@@ -21,7 +21,7 @@ numba_cuda/numba/cuda/decorators.py,sha256=wCLjm2FUWjznmKV_QK1egO03msOrJEnZTR1du
 numba_cuda/numba/cuda/descriptor.py,sha256=t1rSVJSCAlVACC5_Un3FQ7iubdTTBe-euqz88cvs2tI,985
 numba_cuda/numba/cuda/device_init.py,sha256=FwfHZ72aJdeFBfY4TqhBMpEBf2hcA2pnlGRy2Eu37CE,3439
 numba_cuda/numba/cuda/deviceufunc.py,sha256=zj9BbLiZD-dPttHew4olw8ANgR2nXnXEE9qjCeGLrQI,30731
-numba_cuda/numba/cuda/dispatcher.py,sha256=5oW6oYAkNw5WuQ44s7t-iROaySggUViqFBVCqKRJ5qU,53172
+numba_cuda/numba/cuda/dispatcher.py,sha256=QKaZxhYh7EGc-b4Rt4-uvpKR_X2yXerpOud2EU96Pcs,53206
 numba_cuda/numba/cuda/errors.py,sha256=WRso1Q_jCoWP5yrDBMhihRhhVtVo1-7KdN8QVE9j46o,1712
 numba_cuda/numba/cuda/extending.py,sha256=Haz317ITossSxMDaaii2SkXVVj4O2t-3KKZfE3fCxw4,2304
 numba_cuda/numba/cuda/flags.py,sha256=4p12gufxChLfF1fCK_B9HJRdB0Eg8k_UFU__cItqWps,743
@@ -124,7 +124,7 @@ numba_cuda/numba/cuda/tests/core/serialize_usecases.py,sha256=WSWlkvYh_2-vffs0d-
 numba_cuda/numba/cuda/tests/core/test_serialize.py,sha256=iemDgSNscpDjBq1Uy83JI1J-Gv8vzUXuebFXq1B9S7c,11224
 numba_cuda/numba/cuda/tests/cudadrv/__init__.py,sha256=GdfSq6pRVSOQwmgNi7ZFQ5l0yg4-2gNar_0Rz0buUpM,157
 numba_cuda/numba/cuda/tests/cudadrv/test_array_attr.py,sha256=np2UccpkNqLFLcbjJNHGkCJF6TZIyFljB6UXD7U71nM,5299
-numba_cuda/numba/cuda/tests/cudadrv/test_context_stack.py,sha256=8hRDvMyl1ZQaGzPuX0PJV4UXGjFO13vh3PB0jzx_QNM,5623
+numba_cuda/numba/cuda/tests/cudadrv/test_context_stack.py,sha256=lEf4vYeDGhpzR6ppcKaTyjwTWPHj_OO_Rtiz5DK4bDY,5939
 numba_cuda/numba/cuda/tests/cudadrv/test_cuda_array_slicing.py,sha256=QhBQr3ZzrBMT-r132RR99UCKwrR-RwZk98RxVv5os0w,13933
 numba_cuda/numba/cuda/tests/cudadrv/test_cuda_auto_context.py,sha256=xprxASXl0g6QrOujoj07YDw3lIwu0SQbk1lGQPJHlRc,564
 numba_cuda/numba/cuda/tests/cudadrv/test_cuda_devicerecord.py,sha256=JkMbKFa3CBSMSQaSWzOqJU7DE5YlwJLux6OLAmvnSJo,5654
@@ -174,7 +174,7 @@ numba_cuda/numba/cuda/tests/cudapy/test_boolean.py,sha256=j4mIOv4rJTLjJzpKk1O9UF
 numba_cuda/numba/cuda/tests/cudapy/test_caching.py,sha256=0KJ09VzYt4fH-BlBWtMHoCuTQrvdFN3rkD_b6GoEPzg,18166
 numba_cuda/numba/cuda/tests/cudapy/test_casting.py,sha256=3LaN3ZsSuOZXAZXCV85wYyhh0ih7JqABnjGTa7Y2YBE,8748
 numba_cuda/numba/cuda/tests/cudapy/test_cffi.py,sha256=tC7ZCA4dkzehS33iz2l35rX6OxE3BTQd9ivV4r74YXs,926
-numba_cuda/numba/cuda/tests/cudapy/test_compiler.py,sha256=Nk38CufGMCRGhxks33asL63IDANZNYJ4TPOAzqGKD7o,12909
+numba_cuda/numba/cuda/tests/cudapy/test_compiler.py,sha256=XCAg0Ghx5d7P595k3sNoNYSVoVzey9taPsfOVDwLjIo,13364
 numba_cuda/numba/cuda/tests/cudapy/test_complex.py,sha256=nqtRw7FEnyPaE0N-UmyQIvQVSdICSh7b_-VY0vNV4MA,10134
 numba_cuda/numba/cuda/tests/cudapy/test_complex_kernel.py,sha256=KIuXQ0ihgQQXM-eH7s3xAxhKe35YL1qDTHCVTWA4ut8,497
 numba_cuda/numba/cuda/tests/cudapy/test_const_string.py,sha256=li1UsV5vc2M01cJ7k6_526VPtuAOAKr8e7kb1CDUXi4,4323
@@ -201,7 +201,7 @@ numba_cuda/numba/cuda/tests/cudapy/test_gufunc_scalar.py,sha256=vAP2ggp2arBqJS8k
 numba_cuda/numba/cuda/tests/cudapy/test_gufunc_scheduling.py,sha256=1USofSlavYFaVhP8oep5oJ-CLzXxYwkI3EtOkY6jrVw,2610
 numba_cuda/numba/cuda/tests/cudapy/test_idiv.py,sha256=tTy7hN2LJ4897UzO3EUxjuUzbBcs9QITHJu3s_eknq0,1054
 numba_cuda/numba/cuda/tests/cudapy/test_inline.py,sha256=T7DHquV_4HuX5fFQQS3kcZzgifTzwYbMFiY7SgQzoLA,4584
-numba_cuda/numba/cuda/tests/cudapy/test_inspect.py,sha256=M9xPAoPkwEsaqrFJtfWnF225GnShsbYHaxhmcOl2JFQ,6936
+numba_cuda/numba/cuda/tests/cudapy/test_inspect.py,sha256=DxXqAKjnDHPHqFylWGhRXbpSl7Y0C18W8QLs0qIKnGw,7400
 numba_cuda/numba/cuda/tests/cudapy/test_intrinsics.py,sha256=5dy-YHXCFwy7vS-AmW26Dx1hAoKHlqFR54wUNE-vMKY,37414
 numba_cuda/numba/cuda/tests/cudapy/test_ipc.py,sha256=bNT6UZgsgeVWyzBrlKXucQW6IKcD6NEmbwV5cFhf-7I,10553
 numba_cuda/numba/cuda/tests/cudapy/test_iterators.py,sha256=WCRkQfkEnB0d9aj55dVvyQzD4QxrOLubnlKO0xTiNto,2343
@@ -289,8 +289,8 @@ numba_cuda/numba/cuda/tests/test_binary_generation/generate_raw_ltoir.py,sha256=
 numba_cuda/numba/cuda/tests/test_binary_generation/nrt_extern.cu,sha256=T9ubst3fFUK7EXyXXMi73wAban3VFFQ986cY5OcKfvI,157
 numba_cuda/numba/cuda/tests/test_binary_generation/test_device_functions.cu,sha256=IB5t-dVhrKVoue3AbUx3yVMxPG0hBF_yZbzb4642sf0,538
 numba_cuda/numba/cuda/tests/test_binary_generation/undefined_extern.cu,sha256=q3oxZziT8KDodeNcEBiWULH6vMrHCWucmJmtrg8C0d0,128
-numba_cuda-0.18.0.dist-info/licenses/LICENSE,sha256=eHeYE-XjASmwbxfsP5AImgfzRwZurZGqH1f6OFwJ4io,1326
-numba_cuda-0.18.0.dist-info/METADATA,sha256=sXYrBmcXW_sONKfaiXq-scejvjNGjaektP5gmV1F6II,3775
-numba_cuda-0.18.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-numba_cuda-0.18.0.dist-info/top_level.txt,sha256=C50SsH-8tXDmt7I0Y3nlJYhS5s6pqWflCPdobe9vx2M,11
-numba_cuda-0.18.0.dist-info/RECORD,,
+numba_cuda-0.18.1.dist-info/licenses/LICENSE,sha256=eHeYE-XjASmwbxfsP5AImgfzRwZurZGqH1f6OFwJ4io,1326
+numba_cuda-0.18.1.dist-info/METADATA,sha256=zuCl0VNMdrCBmKYgnVcqSmbHl766dAxT1CwXRMT3NmA,3775
+numba_cuda-0.18.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+numba_cuda-0.18.1.dist-info/top_level.txt,sha256=C50SsH-8tXDmt7I0Y3nlJYhS5s6pqWflCPdobe9vx2M,11
+numba_cuda-0.18.1.dist-info/RECORD,,