Skip to content

Commit

Permalink
Document MVC env and config vars
Browse files Browse the repository at this point in the history
  • Loading branch information
brandon-b-miller authored and gmarkall committed Oct 23, 2024
1 parent e11fdf6 commit e5f2ef9
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 2 deletions.
10 changes: 10 additions & 0 deletions docs/source/reference/envvars.rst
Original file line number Diff line number Diff line change
Expand Up @@ -120,4 +120,14 @@ target.
``/usr/local/cuda/include``. On Windows, the default is
``$env:CUDA_PATH\include``.

.. envvar:: NUMBA_CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY

Enable minor version compatibility for the CUDA driver. Requires the
``cubinlinker`` and ``ptxcompiler`` packages to be installed. Provides minor
version compatibility for driver versions less than 12.0.

.. envvar:: NUMBA_CUDA_ENABLE_PYNVJITLINK

Use ``pynvjitlink`` for minor version compatibility. Requires the ``pynvjitlink``
package to be installed. Provides minor version compatibility for driver versions
greater than 12.0.
6 changes: 4 additions & 2 deletions docs/source/user/minor_version_compatibility.rst
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,8 @@ MVC support is enabled by setting the environment variable:

.. code:: bash
export NUMBA_CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY=1
export NUMBA_CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY=1 # CUDA 11
export NUMBA_CUDA_ENABLE_PYNVJITLINK=1 # CUDA 12
or by setting a configuration variable prior to using any CUDA functionality in
Expand All @@ -74,7 +75,8 @@ Numba:
.. code:: python
from numba import config
config.CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY = True
config.CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY = True # CUDA 11
config.CUDA_ENABLE_PYNVJITLINK = True # CUDA 12
References
Expand Down

0 comments on commit e5f2ef9

Please sign in to comment.