numba-cuda 0.0.12__py3-none-any.whl → 0.0.14__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
_numba_cuda_redirector.py CHANGED
@@ -47,8 +47,10 @@ class NumbaCudaFinder(importlib.abc.MetaPathFinder):
47
47
  return False
48
48
 
49
49
  self.numba_path = numba_search_locations[0]
50
- self.numba_cuda_path = str((pathlib.Path(numba_cuda_search_locations[0]) /
51
- 'numba'))
50
+
51
+ location = numba_cuda_search_locations[0]
52
+ self.numba_cuda_path = str((pathlib.Path(location) / 'numba'))
53
+
52
54
  self.initialized = True
53
55
  return True
54
56
 
numba_cuda/VERSION CHANGED
@@ -1 +1 @@
1
- 0.0.12
1
+ 0.0.14
@@ -401,8 +401,8 @@ class atomic(Stub):
401
401
 
402
402
  Performs::
403
403
 
404
- ary[idx] = (value if (array[idx] == 0) or
405
- (array[idx] > value) else array[idx] - 1)
404
+ ary[idx] = (val if (ary[idx] == 0) or
405
+ (ary[idx] > val) else ary[idx] - 1)
406
406
 
407
407
  Supported on uint32, and uint64 operands only.
408
408
 
@@ -87,6 +87,8 @@ class TestNvvmDriver(unittest.TestCase):
87
87
  self.assertIn('section "llvm.metadata"', used_line)
88
88
 
89
89
  def test_nvvm_ir_verify_fail(self):
90
+ if runtime.get_version() >= (12, 5):
91
+ self.skipTest("Bad triple doesn't fail verify on CUDA >= 12.5")
90
92
  m = ir.Module("test_bad_ir")
91
93
  m.triple = "unknown-unknown-unknown"
92
94
  m.data_layout = NVVM().data_layout
@@ -558,18 +558,12 @@ class TestCudaAtomics(CUDATestCase):
558
558
  # Use the first (and only) definition
559
559
  asm = next(iter(kernel.inspect_asm().values()))
560
560
  if cc_X_or_above(6, 0):
561
- if cuda.runtime.get_version() > (12, 1):
562
- # CUDA 12.2 and above generate a more optimized reduction
563
- # instruction, because the result does not need to be
564
- # placed in a register.
565
- inst = 'red'
566
- else:
567
- inst = 'atom'
561
+ inst = "(red|atom)"
568
562
 
569
563
  if shared:
570
- inst = f'{inst}.shared'
564
+ inst = f'{inst}\\.shared'
571
565
 
572
- self.assertIn(f'{inst}.add.f64', asm)
566
+ self.assertRegex(asm, f'{inst}.add.f64', asm)
573
567
  else:
574
568
  if shared:
575
569
  self.assertIn('atom.shared.cas.b64', asm)
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: numba-cuda
3
- Version: 0.0.12
3
+ Version: 0.0.14
4
4
  Summary: CUDA target for Numba
5
5
  Author: Anaconda Inc., NVIDIA Corporation
6
6
  License: BSD 2-clause
@@ -11,6 +11,7 @@ Project-URL: License, https://github.com/rapidsai/numba-cuda/blob/main/LICENSE
11
11
  Requires-Python: >=3.9
12
12
  Description-Content-Type: text/markdown
13
13
  License-File: LICENSE
14
+ Requires-Dist: numba>=0.59.1
14
15
 
15
16
  # Numba CUDA Target
16
17
 
@@ -1,6 +1,6 @@
1
1
  _numba_cuda_redirector.pth,sha256=cmfMMmV0JPh3yEpl4bGeM9AuXiVVMSo6Z_b7RaQL3XE,30
2
- _numba_cuda_redirector.py,sha256=qzl4_6GJW4G0n4xD3o03_PRzVUdR2ASR4RX-V0M-OgY,2407
3
- numba_cuda/VERSION,sha256=xOsNOougYJfHhO5ums87sktNAek9fFpZas6Hsj6jAVA,7
2
+ _numba_cuda_redirector.py,sha256=rc56rnb40w3AtrqnhS66JSgYTSTsi3iTn8yP3NuoQV8,2401
3
+ numba_cuda/VERSION,sha256=D2Ozgb2Qx-omyy1nDcowrtDQ3fcqNTYj5zIC8yKuxlc,7
4
4
  numba_cuda/__init__.py,sha256=atXeUvJKR3JHcAiCFbXCVOJQUHgB1TulmsqSL_9RT3Q,114
5
5
  numba_cuda/_version.py,sha256=jbdUsbR7sVllw0KxQNB0-FMd929CGg3kH2fhHdrlkuc,719
6
6
  numba_cuda/numba/cuda/__init__.py,sha256=idyVHOObC9lTYnp62v7rVprSacRM4d5F6vhXfG5ElTI,621
@@ -37,7 +37,7 @@ numba_cuda/numba/cuda/nvvmutils.py,sha256=W1zr1TpnmFjTkHF0qeu5wnBHub6gzrnpzsvgmu
37
37
  numba_cuda/numba/cuda/printimpl.py,sha256=Eg_RvmOcmFeL6s2RlTD-oOxebGdYLN9VZM1gcY3YMms,2789
38
38
  numba_cuda/numba/cuda/random.py,sha256=khX8iDdde_RTUPWhAqrxZacHRQAorFr7BokPuxRWzrg,10456
39
39
  numba_cuda/numba/cuda/simulator_init.py,sha256=W_bPRtmPGOQVuiprbgt7ENnnnELv_LPCeLDIsfsvFZ8,460
40
- numba_cuda/numba/cuda/stubs.py,sha256=a9smsDwbimTKy0YnV88OkcbYqldMgkx-WiV8LB4-MGs,22280
40
+ numba_cuda/numba/cuda/stubs.py,sha256=W3tozv4ganMnfbdFqyPjgQXYeX8GQhwx_xXgv8jk6iM,22270
41
41
  numba_cuda/numba/cuda/target.py,sha256=EI6XuKQeqvng0uSx_V9jDoxbgFivqSz-4jczFzAbs5o,16837
42
42
  numba_cuda/numba/cuda/testing.py,sha256=E0wP2vfno1yWsl0v1zg31kpbU8FrKxTF-5y9Iv4WjA4,6412
43
43
  numba_cuda/numba/cuda/types.py,sha256=WVfjcly_VUpG9FfKueiEPzZm2NV8Hg0XAFg3bNzPdVc,1314
@@ -100,7 +100,7 @@ numba_cuda/numba/cuda/tests/cudadrv/test_is_fp16.py,sha256=0KPe4E9wOZsSV_0QI0Lmj
100
100
  numba_cuda/numba/cuda/tests/cudadrv/test_linker.py,sha256=_l2_EQEko2Jet5ooj4XMT0L4BjOuqLjbONGj1_MVI50,10161
101
101
  numba_cuda/numba/cuda/tests/cudadrv/test_managed_alloc.py,sha256=kYXYMkx_3GPAITKp4reLeM8KSzKkpxiC8nxnBvXpaTA,4979
102
102
  numba_cuda/numba/cuda/tests/cudadrv/test_mvc.py,sha256=984jATSa01SRoSrVqxPeO6ujJ7w2jsnZa39ABInFLVI,1529
103
- numba_cuda/numba/cuda/tests/cudadrv/test_nvvm_driver.py,sha256=VTw8-TZ7DvjNEqsrzZ-RKKzLRiSKEyAAZ3pRDYEQiVk,7252
103
+ numba_cuda/numba/cuda/tests/cudadrv/test_nvvm_driver.py,sha256=DF7KV5uh-yMztks0f47NhpalV64dvsNy-f8HY6GhAhE,7373
104
104
  numba_cuda/numba/cuda/tests/cudadrv/test_pinned.py,sha256=u_TthSS2N-2J4eBIuF4PGg33AjD-wxly7MKpz0vRAKc,944
105
105
  numba_cuda/numba/cuda/tests/cudadrv/test_profiler.py,sha256=MQWZx1j3lbEpWmIpQ1bV9szrGOV3VHN0QrEnJRjAhW4,508
106
106
  numba_cuda/numba/cuda/tests/cudadrv/test_ptds.py,sha256=S8y-j_ZB4Y_r2S5GGtfFxfracnJflQAhHYfhybKIcUE,4949
@@ -118,7 +118,7 @@ numba_cuda/numba/cuda/tests/cudapy/test_alignment.py,sha256=dik8i4fG6MPlxVilW4l9
118
118
  numba_cuda/numba/cuda/tests/cudapy/test_array.py,sha256=bS6rzvp6BKVLFyW8mFRbVoZbxIbc2WCl5SzQ6XG0s8c,10515
119
119
  numba_cuda/numba/cuda/tests/cudapy/test_array_args.py,sha256=XTX4cT7BZexmw0BZPzeudf4OZgM6GNqzjDPyIxJyTdk,4979
120
120
  numba_cuda/numba/cuda/tests/cudapy/test_array_methods.py,sha256=shdeSAOKaoZbrvC8hXhETWH8FhyZPTmHg_TMw2DcdUA,969
121
- numba_cuda/numba/cuda/tests/cudapy/test_atomics.py,sha256=NvZ2XxQMa6Psyg9PQAFYcdYMOr9OjhGMYvYD3Vp5WiI,58449
121
+ numba_cuda/numba/cuda/tests/cudapy/test_atomics.py,sha256=yQWTHQH7WPafLwNhnfOWqAskybXTw1BBwvxL5OLqEAk,58177
122
122
  numba_cuda/numba/cuda/tests/cudapy/test_blackscholes.py,sha256=dpPWyCnxRUThaK6npGvN5_fhU-KT1b1GifU5Tj3AHMI,4023
123
123
  numba_cuda/numba/cuda/tests/cudapy/test_boolean.py,sha256=S8fNBS4gVEZyKk6PpeF5GfG0TLEu_0NvTgybyPNNGLU,547
124
124
  numba_cuda/numba/cuda/tests/cudapy/test_caching.py,sha256=5o1VcVbir1Lro6yFCerbfbjsxH9xNsYFl7N4Rud6ntw,19082
@@ -224,8 +224,8 @@ numba_cuda/numba/cuda/tests/nocuda/test_function_resolution.py,sha256=o4DYocyHK7
224
224
  numba_cuda/numba/cuda/tests/nocuda/test_import.py,sha256=teiL8rpFGQOh41kyBSSNHHFYAJYgpdStXkTcpK4_fxo,1641
225
225
  numba_cuda/numba/cuda/tests/nocuda/test_library_lookup.py,sha256=7kJOPHEcrjy_kTA9Ym-iT_B972bgFRu3UkRtwIgWtuI,7948
226
226
  numba_cuda/numba/cuda/tests/nocuda/test_nvvm.py,sha256=n0_-xFaw6QqiZbhe55oy7lnEeOwqTvA55p5EUFiTpNw,2006
227
- numba_cuda-0.0.12.dist-info/LICENSE,sha256=eHeYE-XjASmwbxfsP5AImgfzRwZurZGqH1f6OFwJ4io,1326
228
- numba_cuda-0.0.12.dist-info/METADATA,sha256=EdnZn8O4Nz8jnYbi2gF1u7_ngp5gSn_oh0Uo4d8rgJs,2027
229
- numba_cuda-0.0.12.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
230
- numba_cuda-0.0.12.dist-info/top_level.txt,sha256=C50SsH-8tXDmt7I0Y3nlJYhS5s6pqWflCPdobe9vx2M,11
231
- numba_cuda-0.0.12.dist-info/RECORD,,
227
+ numba_cuda-0.0.14.dist-info/LICENSE,sha256=eHeYE-XjASmwbxfsP5AImgfzRwZurZGqH1f6OFwJ4io,1326
228
+ numba_cuda-0.0.14.dist-info/METADATA,sha256=qDJCNjO0Yc0aVBixkzNBrBFdEA-KNyQeomXGCV9yaJ8,2056
229
+ numba_cuda-0.0.14.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
230
+ numba_cuda-0.0.14.dist-info/top_level.txt,sha256=C50SsH-8tXDmt7I0Y3nlJYhS5s6pqWflCPdobe9vx2M,11
231
+ numba_cuda-0.0.14.dist-info/RECORD,,