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 CHANGED
@@ -1 +1 @@
1
- 0.15.1
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 if the environment variables NUMBA_CUDA_ENABLE_PYNVJITLINK
7
- # or CUDA_ENABLE_PYNVJITLINK are set, or if the pynvjitlink module is found. If
8
- # explicitly disabled, do not use pynvjitlink, even if present in the env.
9
- _pynvjitlink_enabled_in_env = _readenv(
10
- "NUMBA_CUDA_ENABLE_PYNVJITLINK", bool, None
11
- )
12
- _pynvjitlink_enabled_in_cfg = getattr(config, "CUDA_ENABLE_PYNVJITLINK", None)
13
-
14
- if _pynvjitlink_enabled_in_env is not None:
15
- ENABLE_PYNVJITLINK = _pynvjitlink_enabled_in_env
16
- elif _pynvjitlink_enabled_in_cfg is not None:
17
- ENABLE_PYNVJITLINK = _pynvjitlink_enabled_in_cfg
18
- else:
19
- ENABLE_PYNVJITLINK = importlib.util.find_spec("pynvjitlink") is not None
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 importlib.util.find_spec("cuda.bindings"):
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.1
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
- ```shell
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=RKv2awImAXmzpQmw8QtfJvQx8xbvTKUPGHZMnkuM_Uo,7
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=C4GaDax1WYQgj6JaPkHAHXSxI9grkEigbrTW03foqKU,2369
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.1.dist-info/licenses/LICENSE,sha256=eHeYE-XjASmwbxfsP5AImgfzRwZurZGqH1f6OFwJ4io,1326
278
- numba_cuda-0.15.1.dist-info/METADATA,sha256=tIHuoF1LHBDRmxYFwmcx6Lgv3Te5Pl539NwfRn21yLk,2907
279
- numba_cuda-0.15.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
280
- numba_cuda-0.15.1.dist-info/top_level.txt,sha256=C50SsH-8tXDmt7I0Y3nlJYhS5s6pqWflCPdobe9vx2M,11
281
- numba_cuda-0.15.1.dist-info/RECORD,,
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,,