Skip to content

CUDA: Add CUDA 11.8 / Hopper support and required fixes #8579

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
esc merged 5 commits into from
Dec 3, 2022
Merged

CUDA: Add CUDA 11.8 / Hopper support and required fixes #8579

Changes from 1 commit
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
4a9672f
CUDA: Fix supported CC logic
gmarkall Nov 4, 2022
f7adcec
CUDA tests: always add the NVVM IR version
gmarkall Nov 4, 2022
e3e7d82
NVVM: Don't cache supported CCs at global scope
gmarkall Nov 14, 2022
c4e4081
Clarify that supported CC bounds are inclusive
gmarkall Nov 14, 2022
3e7ebb4
NVVM: Don't use hardcoded version limits
gmarkall Nov 14, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
NVVM: Don't use hardcoded version limits 
The minimum toolkit version can be derived from the support matrix, and
the minimum CC for unrecognised toolkits can be derived from
`config.CUDA_DEFAULT_PTX_CC`, which can also be overridden by the user
if necessary - for example, if a user has a toolkit that in future has
dropped even support for 5.3.
  • Loading branch information
@gmarkall
gmarkall committed Nov 14, 2022
commit 3e7ebb47d911d103845c55c7e3ed26dd6f9c7919
9 changes: 6 additions & 3 deletions numba/cuda/cudadrv/nvvm.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -380,7 +380,8 @@ def ccs_supported_by_ctk(ctk_version):
except KeyError:
# For unsupported CUDA toolkit versions, all we can do is assume all
# non-deprecated versions we are aware of are supported.
return tuple([cc for cc in COMPUTE_CAPABILITIES if cc >= (5, 3)])
return tuple([cc for cc in COMPUTE_CAPABILITIES
if cc >= config.CUDA_DEFAULT_PTX_CC])


def get_supported_ccs():
Expand All @@ -394,11 +395,13 @@ def get_supported_ccs():
return _supported_cc

# Ensure the minimum CTK version requirement is met
if cudart_version < (11, 0):
min_cudart = min(CTK_SUPPORTED)
if cudart_version < min_cudart:
_supported_cc = ()
ctk_ver = f"{cudart_version[0]}.{cudart_version[1]}"
unsupported_ver = (f"CUDA Toolkit {ctk_ver} is unsupported by Numba - "
"11.0 is the minimum required version.")
f"{min_cudart[0]}.{min_cudart[1]} is the minimum "
"required version.")
warnings.warn(unsupported_ver)
return _supported_cc

Expand Down