From 2da434f81a9dc8c3ad965bf33820ea850bfbf8d2 Mon Sep 17 00:00:00 2001 From: Xudong Zhang Date: Tue, 8 Aug 2023 16:17:59 +0800 Subject: [PATCH 1/2] set default coompute capability according to cuda version (#773) --- setup.py | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) diff --git a/setup.py b/setup.py index 2644c03b1637..00470640fec5 100644 --- a/setup.py +++ b/setup.py @@ -47,12 +47,6 @@ def get_nvcc_cuda_version(cuda_dir: str) -> Version: raise RuntimeError( "GPUs with compute capability less than 7.0 are not supported.") compute_capabilities.add(major * 10 + minor) -# If no GPU is available, add all supported compute capabilities. -if not compute_capabilities: - compute_capabilities = {70, 75, 80, 86, 90} -# Add target compute capabilities to NVCC flags. -for capability in compute_capabilities: - NVCC_FLAGS += ["-gencode", f"arch=compute_{capability},code=sm_{capability}"] # Validate the NVCC CUDA version. nvcc_cuda_version = get_nvcc_cuda_version(CUDA_HOME) @@ -65,6 +59,18 @@ def get_nvcc_cuda_version(cuda_dir: str) -> Version: raise RuntimeError( "CUDA 11.8 or higher is required for GPUs with compute capability 9.0.") +# If no GPU is available, add all supported compute capabilities. +if not compute_capabilities: + compute_capabilities = {70, 75} + if nvcc_cuda_version >= Version("11.1"): + compute_capabilities.add(86) + if nvcc_cuda_version >= Version("11.8"): + compute_capabilities.add(90) + +# Add target compute capabilities to NVCC flags. +for capability in compute_capabilities: + NVCC_FLAGS += ["-gencode", f"arch=compute_{capability},code=sm_{capability}"] + # Use NVCC threads to parallelize the build. if nvcc_cuda_version >= Version("11.2"): num_threads = min(os.cpu_count(), 8) From 7c4fdcfc9766a6ee6b08de5a42f413c792453a77 Mon Sep 17 00:00:00 2001 From: Zhuohan Li Date: Mon, 21 Aug 2023 23:04:26 +0000 Subject: [PATCH 2/2] Add compute capability 80 --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 00470640fec5..8cb0d7bbbd68 100644 --- a/setup.py +++ b/setup.py @@ -61,7 +61,7 @@ def get_nvcc_cuda_version(cuda_dir: str) -> Version: # If no GPU is available, add all supported compute capabilities. if not compute_capabilities: - compute_capabilities = {70, 75} + compute_capabilities = {70, 75, 80} if nvcc_cuda_version >= Version("11.1"): compute_capabilities.add(86) if nvcc_cuda_version >= Version("11.8"):