diff options
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: |