diff options
| author | Olexa Bilaniuk <obilaniu@gmail.com> | 2022-11-28 20:07:14 -0500 |
|---|---|---|
| committer | Eli Schwartz <eschwartz93@gmail.com> | 2022-11-29 18:44:40 -0500 |
| commit | c4f35892156b33e6f9d96577ff62df8e77a4aa02 (patch) | |
| tree | 3fc8bf5235d67cd88a0434c8b360228461f28ad1 /mesonbuild/modules/cuda.py | |
| parent | 1a6a014494785e107023b25c157b30c3a4838b99 (diff) | |
| download | meson-c4f35892156b33e6f9d96577ff62df8e77a4aa02.zip meson-c4f35892156b33e6f9d96577ff62df8e77a4aa02.tar.gz meson-c4f35892156b33e6f9d96577ff62df8e77a4aa02.tar.bz2 | |
CUDA: Add Orin, Lovelace and Hopper architecture names for CUDA 11.8
Co-authored-by: David Seifert <soap@gentoo.org>
Diffstat (limited to 'mesonbuild/modules/cuda.py')
| -rw-r--r-- | mesonbuild/modules/cuda.py | 18 |
1 files changed, 15 insertions, 3 deletions
diff --git a/mesonbuild/modules/cuda.py b/mesonbuild/modules/cuda.py index 0e33984..d3ff992 100644 --- a/mesonbuild/modules/cuda.py +++ b/mesonbuild/modules/cuda.py @@ -246,11 +246,20 @@ class CudaModule(NewExtensionModule): cuda_hi_limit_gpu_architecture = '8.6' # noqa: E221 if version_compare(cuda_version, '>=11.1'): - cuda_common_gpu_architectures += ['8.6', '8.6+PTX'] # noqa: E221 + cuda_common_gpu_architectures += ['8.6'] # noqa: E221 cuda_all_gpu_architectures += ['8.6'] # noqa: E221 - if version_compare(cuda_version, '<12.0'): - cuda_hi_limit_gpu_architecture = '9.0' # noqa: E221 + if version_compare(cuda_version, '<11.8'): + cuda_common_gpu_architectures += ['8.6+PTX'] # noqa: E221 + cuda_hi_limit_gpu_architecture = '8.7' # noqa: E221 + + if version_compare(cuda_version, '>=11.8'): + cuda_known_gpu_architectures += ['Orin', 'Lovelace', 'Hopper'] # noqa: E221 + cuda_common_gpu_architectures += ['8.9', '9.0', '9.0+PTX'] # noqa: E221 + cuda_all_gpu_architectures += ['8.7', '8.9', '9.0'] # noqa: E221 + + if version_compare(cuda_version, '<12'): + cuda_hi_limit_gpu_architecture = '9.1' # noqa: E221 if not cuda_arch_list: cuda_arch_list = 'Auto' @@ -301,6 +310,9 @@ class CudaModule(NewExtensionModule): 'Xavier': (['7.2'], []), 'Turing': (['7.5'], ['7.5']), 'Ampere': (['8.0'], ['8.0']), + 'Orin': (['8.7'], []), + 'Lovelace': (['8.9'], ['8.9']), + 'Hopper': (['9.0'], ['9.0']), }.get(arch_name, (None, None)) if arch_bin is None: |