numba-cuda 0.0.13__py3-none-any.whl → 0.0.15__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 +4 -2
- numba_cuda/VERSION +1 -1
- numba_cuda/numba/cuda/stubs.py +2 -2
- numba_cuda/numba/cuda/tests/cudadrv/test_nvvm_driver.py +2 -0
- numba_cuda/numba/cuda/tests/cudapy/test_atomics.py +3 -9
- {numba_cuda-0.0.13.dist-info → numba_cuda-0.0.15.dist-info}/METADATA +2 -2
- {numba_cuda-0.0.13.dist-info → numba_cuda-0.0.15.dist-info}/RECORD +10 -10
- {numba_cuda-0.0.13.dist-info → numba_cuda-0.0.15.dist-info}/WHEEL +1 -1
- {numba_cuda-0.0.13.dist-info → numba_cuda-0.0.15.dist-info}/LICENSE +0 -0
- {numba_cuda-0.0.13.dist-info → numba_cuda-0.0.15.dist-info}/top_level.txt +0 -0
_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
|
-
|
51
|
-
|
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.
|
1
|
+
0.0.15
|
numba_cuda/numba/cuda/stubs.py
CHANGED
@@ -401,8 +401,8 @@ class atomic(Stub):
|
|
401
401
|
|
402
402
|
Performs::
|
403
403
|
|
404
|
-
ary[idx] = (
|
405
|
-
(
|
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
|
-
|
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}
|
564
|
+
inst = f'{inst}\\.shared'
|
571
565
|
|
572
|
-
self.
|
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.
|
3
|
+
Version: 0.0.15
|
4
4
|
Summary: CUDA target for Numba
|
5
5
|
Author: Anaconda Inc., NVIDIA Corporation
|
6
6
|
License: BSD 2-clause
|
@@ -11,7 +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
|
14
|
+
Requires-Dist: numba>=0.59.1
|
15
15
|
|
16
16
|
# Numba CUDA Target
|
17
17
|
|
@@ -1,6 +1,6 @@
|
|
1
1
|
_numba_cuda_redirector.pth,sha256=cmfMMmV0JPh3yEpl4bGeM9AuXiVVMSo6Z_b7RaQL3XE,30
|
2
|
-
_numba_cuda_redirector.py,sha256=
|
3
|
-
numba_cuda/VERSION,sha256=
|
2
|
+
_numba_cuda_redirector.py,sha256=rc56rnb40w3AtrqnhS66JSgYTSTsi3iTn8yP3NuoQV8,2401
|
3
|
+
numba_cuda/VERSION,sha256=x6dzGoPsZ9EHtlIygol3rcQ3OvE33CLyeLWdZ0e3nuI,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=
|
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=
|
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=
|
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.
|
228
|
-
numba_cuda-0.0.
|
229
|
-
numba_cuda-0.0.
|
230
|
-
numba_cuda-0.0.
|
231
|
-
numba_cuda-0.0.
|
227
|
+
numba_cuda-0.0.15.dist-info/LICENSE,sha256=eHeYE-XjASmwbxfsP5AImgfzRwZurZGqH1f6OFwJ4io,1326
|
228
|
+
numba_cuda-0.0.15.dist-info/METADATA,sha256=cirF9huDCXInV_t0pINibeeFL8lN3oTQreltB_vx1sQ,2056
|
229
|
+
numba_cuda-0.0.15.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
|
230
|
+
numba_cuda-0.0.15.dist-info/top_level.txt,sha256=C50SsH-8tXDmt7I0Y3nlJYhS5s6pqWflCPdobe9vx2M,11
|
231
|
+
numba_cuda-0.0.15.dist-info/RECORD,,
|
File without changes
|
File without changes
|