Skip to content

Commit

Permalink
give suggesitions
Browse files Browse the repository at this point in the history
  • Loading branch information
isVoid committed Feb 18, 2025
1 parent a1fdff6 commit c72d790
Showing 1 changed file with 9 additions and 3 deletions.
12 changes: 9 additions & 3 deletions pynvjitlink/patch.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,12 +9,10 @@
_numba_version_ok = False
_numba_error = None

_numba_cuda_version_ok = False
_numba_cuda_in_use = False
_numba_cuda_error = None

required_numba_ver = (0, 58)
max_numba_cuda_ver = (0, 0, 18)

mvc_docs_url = (
"https://numba.readthedocs.io/en/stable/cuda/" "minor_version_compatibility.html"
Expand Down Expand Up @@ -53,7 +51,15 @@
spec = importlib.util.find_spec("numba_cuda")
if spec is not None:
_numba_cuda_in_use = True
_numba_cuda_error = "`numba_cuda` includes patches from pynvjitlink, so no further patches are needed."
_numba_cuda_error = "`numba_cuda` includes patches from pynvjitlink, so no further patches are needed. "

import numba_cuda

numba_cuda_ver = (int(x) for x in numba_cuda.__version__.split("."))
if numba_cuda_ver < (0, 2, 0):
suggestion = "Instead, use NUMBA_CUDA_ENABLE_PYNVJITLINK environment variable to enable pynvjitlink features."
else:
suggestion = "Instead, use config.CUDA_ENABLE_PYNVJITLINK option to enable pynvjitlink features."
else:
_numba_cuda_in_use = False

Expand Down

0 comments on commit c72d790

Please sign in to comment.