Skip to content

Commit 21e0354

Browse files
brandon-b-millergmarkall
authored andcommitted
Document MVC env and config vars
1 parent 126c386 commit 21e0354

File tree

2 files changed

+14
-2
lines changed

2 files changed

+14
-2
lines changed

docs/source/reference/envvars.rst

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -120,4 +120,14 @@ target.
120120
``/usr/local/cuda/include``. On Windows, the default is
121121
``$env:CUDA_PATH\include``.
122122

123+
.. envvar:: NUMBA_CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY
123124

125+
Enable minor version compatibility for the CUDA driver. Requires the
126+
``cubinlinker`` and ``ptxcompiler`` packages to be installed. Provides minor
127+
version compatibility for driver versions less than 12.0.
128+
129+
.. envvar:: NUMBA_CUDA_ENABLE_PYNVJITLINK
130+
131+
Use ``pynvjitlink`` for minor version compatibility. Requires the ``pynvjitlink``
132+
package to be installed. Provides minor version compatibility for driver versions
133+
greater than 12.0.

docs/source/user/minor_version_compatibility.rst

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,8 @@ MVC support is enabled by setting the environment variable:
6565

6666
.. code:: bash
6767
68-
export NUMBA_CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY=1
68+
export NUMBA_CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY=1 # CUDA 11
69+
export NUMBA_CUDA_ENABLE_PYNVJITLINK=1 # CUDA 12
6970
7071
7172
or by setting a configuration variable prior to using any CUDA functionality in
@@ -74,7 +75,8 @@ Numba:
7475
.. code:: python
7576
7677
from numba import config
77-
config.CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY = True
78+
config.CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY = True # CUDA 11
79+
config.CUDA_ENABLE_PYNVJITLINK = True # CUDA 12
7880
7981
8082
References

0 commit comments

Comments
 (0)