numba-cuda 0.15.1__py3-none-any.whl → 0.15.2__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/VERSION +1 -1
- numba_cuda/numba/cuda/__init__.py +19 -18
- {numba_cuda-0.15.1.dist-info → numba_cuda-0.15.2.dist-info}/METADATA +5 -10
- {numba_cuda-0.15.1.dist-info → numba_cuda-0.15.2.dist-info}/RECORD +7 -7
- {numba_cuda-0.15.1.dist-info → numba_cuda-0.15.2.dist-info}/WHEEL +0 -0
- {numba_cuda-0.15.1.dist-info → numba_cuda-0.15.2.dist-info}/licenses/LICENSE +0 -0
- {numba_cuda-0.15.1.dist-info → numba_cuda-0.15.2.dist-info}/top_level.txt +0 -0
numba_cuda/VERSION
CHANGED
@@ -1 +1 @@
|
|
1
|
-
0.15.
|
1
|
+
0.15.2
|
@@ -3,23 +3,21 @@ from numba import runtests
|
|
3
3
|
from numba.core import config
|
4
4
|
from .utils import _readenv
|
5
5
|
|
6
|
-
# Enable pynvjitlink
|
7
|
-
#
|
8
|
-
#
|
9
|
-
|
10
|
-
|
11
|
-
|
12
|
-
|
13
|
-
|
14
|
-
|
15
|
-
|
16
|
-
|
17
|
-
|
18
|
-
|
19
|
-
|
20
|
-
|
21
|
-
if not hasattr(config, "CUDA_ENABLE_PYNVJITLINK"):
|
22
|
-
config.CUDA_ENABLE_PYNVJITLINK = ENABLE_PYNVJITLINK
|
6
|
+
# Enable pynvjitlink based on the following precedence:
|
7
|
+
# 1. Config setting "CUDA_ENABLE_PYNVJITLINK" (highest priority)
|
8
|
+
# 2. Environment variable "NUMBA_CUDA_ENABLE_PYNVJITLINK"
|
9
|
+
# 3. Auto-detection of pynvjitlink module (lowest priority)
|
10
|
+
if getattr(config, "CUDA_ENABLE_PYNVJITLINK", None) is None:
|
11
|
+
if (
|
12
|
+
_pynvjitlink_enabled_in_env := _readenv(
|
13
|
+
"NUMBA_CUDA_ENABLE_PYNVJITLINK", bool, None
|
14
|
+
)
|
15
|
+
) is not None:
|
16
|
+
config.CUDA_ENABLE_PYNVJITLINK = _pynvjitlink_enabled_in_env
|
17
|
+
else:
|
18
|
+
config.CUDA_ENABLE_PYNVJITLINK = (
|
19
|
+
importlib.util.find_spec("pynvjitlink") is not None
|
20
|
+
)
|
23
21
|
|
24
22
|
# Upstream numba sets CUDA_USE_NVIDIA_BINDING to 0 by default, so it always
|
25
23
|
# exists. Override, but not if explicitly set to 0 in the envioronment.
|
@@ -32,7 +30,10 @@ else:
|
|
32
30
|
USE_NV_BINDING = True
|
33
31
|
config.CUDA_USE_NVIDIA_BINDING = USE_NV_BINDING
|
34
32
|
if config.CUDA_USE_NVIDIA_BINDING:
|
35
|
-
if not
|
33
|
+
if not (
|
34
|
+
importlib.util.find_spec("cuda")
|
35
|
+
and importlib.util.find_spec("cuda.bindings")
|
36
|
+
):
|
36
37
|
raise ImportError(
|
37
38
|
"CUDA bindings not found. Please pip install the "
|
38
39
|
"cuda-bindings package. Alternatively, install "
|
@@ -1,6 +1,6 @@
|
|
1
1
|
Metadata-Version: 2.4
|
2
2
|
Name: numba-cuda
|
3
|
-
Version: 0.15.
|
3
|
+
Version: 0.15.2
|
4
4
|
Summary: CUDA target for Numba
|
5
5
|
Author: Anaconda Inc., NVIDIA Corporation
|
6
6
|
License: BSD 2-clause
|
@@ -51,17 +51,10 @@ tracker](https://github.com/NVIDIA/numba-cuda/issues).
|
|
51
51
|
To raise questions or initiate discussions, please use the [Numba Discourse
|
52
52
|
forum](https://numba.discourse.group).
|
53
53
|
|
54
|
-
## Installation with pip
|
54
|
+
## Installation with pip or conda
|
55
55
|
|
56
|
-
|
57
|
-
pip install numba-cuda
|
58
|
-
```
|
59
|
-
|
60
|
-
## Installation with Conda
|
56
|
+
Please refer to the [Installation documentation](https://nvidia.github.io/numba-cuda/user/installation.html#installation-with-a-python-package-manager).
|
61
57
|
|
62
|
-
```shell
|
63
|
-
conda install -c conda-forge numba-cuda
|
64
|
-
```
|
65
58
|
|
66
59
|
## Installation from source
|
67
60
|
|
@@ -71,6 +64,8 @@ Install as an editable install:
|
|
71
64
|
pip install -e .
|
72
65
|
```
|
73
66
|
|
67
|
+
If you want to manage all run-time dependencies yourself, also pass the `--no-deps` flag.
|
68
|
+
|
74
69
|
## Running tests
|
75
70
|
|
76
71
|
```
|
@@ -1,9 +1,9 @@
|
|
1
1
|
_numba_cuda_redirector.pth,sha256=cmfMMmV0JPh3yEpl4bGeM9AuXiVVMSo6Z_b7RaQL3XE,30
|
2
2
|
_numba_cuda_redirector.py,sha256=n_r8MYbu5-vcXMnLJW147k8DnFXXvgb7nPIXnlXwTyQ,2659
|
3
|
-
numba_cuda/VERSION,sha256=
|
3
|
+
numba_cuda/VERSION,sha256=NVouFPan8ss4fZ07daNVCYDQIO4dAFsTm2IpDBZFBoo,7
|
4
4
|
numba_cuda/__init__.py,sha256=atXeUvJKR3JHcAiCFbXCVOJQUHgB1TulmsqSL_9RT3Q,114
|
5
5
|
numba_cuda/_version.py,sha256=nzrrJXi85d18m6SPdsPsetJNClDETkmF1MrEhGLYDBs,734
|
6
|
-
numba_cuda/numba/cuda/__init__.py,sha256=
|
6
|
+
numba_cuda/numba/cuda/__init__.py,sha256=LOFNwCa4pRk8VkzC2yPUHHhRCYe7Now9mR_m5eEZ0mA,2277
|
7
7
|
numba_cuda/numba/cuda/api.py,sha256=mkbZBcBfm819kCywQbH8jAvUex2m4pYTcFD-LE-tXsQ,17638
|
8
8
|
numba_cuda/numba/cuda/api_util.py,sha256=jK8oUD3zf_D5IX7vbjc3uY_5kmOxwgEqO2m_lDHdWfM,861
|
9
9
|
numba_cuda/numba/cuda/args.py,sha256=UlTHTJpwPeCtnW0Bb-Wetm5UO9TPR-PCgIt5ys8b8tQ,1894
|
@@ -274,8 +274,8 @@ numba_cuda/numba/cuda/tests/test_binary_generation/generate_raw_ltoir.py,sha256=
|
|
274
274
|
numba_cuda/numba/cuda/tests/test_binary_generation/nrt_extern.cu,sha256=T9ubst3fFUK7EXyXXMi73wAban3VFFQ986cY5OcKfvI,157
|
275
275
|
numba_cuda/numba/cuda/tests/test_binary_generation/test_device_functions.cu,sha256=IB5t-dVhrKVoue3AbUx3yVMxPG0hBF_yZbzb4642sf0,538
|
276
276
|
numba_cuda/numba/cuda/tests/test_binary_generation/undefined_extern.cu,sha256=q3oxZziT8KDodeNcEBiWULH6vMrHCWucmJmtrg8C0d0,128
|
277
|
-
numba_cuda-0.15.
|
278
|
-
numba_cuda-0.15.
|
279
|
-
numba_cuda-0.15.
|
280
|
-
numba_cuda-0.15.
|
281
|
-
numba_cuda-0.15.
|
277
|
+
numba_cuda-0.15.2.dist-info/licenses/LICENSE,sha256=eHeYE-XjASmwbxfsP5AImgfzRwZurZGqH1f6OFwJ4io,1326
|
278
|
+
numba_cuda-0.15.2.dist-info/METADATA,sha256=cgNS-opTQXpoUK69QTWTgjnaDM96m-mOLv_w2y9fYNk,3043
|
279
|
+
numba_cuda-0.15.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
280
|
+
numba_cuda-0.15.2.dist-info/top_level.txt,sha256=C50SsH-8tXDmt7I0Y3nlJYhS5s6pqWflCPdobe9vx2M,11
|
281
|
+
numba_cuda-0.15.2.dist-info/RECORD,,
|
File without changes
|
File without changes
|
File without changes
|