From 28fbf1e1f30c66716960d5c84430c9cb3289b806 Mon Sep 17 00:00:00 2001 From: Ryker Fish Date: Fri, 11 Jul 2025 14:09:22 -0600 Subject: [PATCH 1/2] changes to CUDA_ARCH --- CMakeLists.txt | 2 +- setup.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 59eedadb..1bbcb019 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -17,7 +17,7 @@ set(CMAKE_CUDA_STANDARD_REQUIRED ON) set(CMAKE_CUDA_SEPARABLE_COMPILATION OFF) # Set CUDA archs so all supported GPUs are covered if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES) - set(CMAKE_CUDA_ARCHITECTURES "all") + set(CMAKE_CUDA_ARCHITECTURES "all-major") endif() IF (CMAKE_BUILD_TYPE MATCHES "Debug") if(CMAKE_CUDA_COMPILER_ID STREQUAL "NVIDIA") diff --git a/setup.py b/setup.py index 024ffd58..195f71d4 100644 --- a/setup.py +++ b/setup.py @@ -53,6 +53,7 @@ def build_extension(self, ext): "-DPYTHON_EXECUTABLE=" + sys.executable, "-DCMAKE_BUILD_TYPE=" + ("Debug" if self.debug else "Release"), f"-DDOUBLEPRECISION={double_precision}", + f"-DCMAKE_CUDA_ARCHITECTURES={os.environ.get('CMAKE_CUDA_ARCHITECTURES', '')}", ] num_jobs = os.cpu_count() From 6bdb35dc8d90b03b76a4e86380b7747807ee054a Mon Sep 17 00:00:00 2001 From: Ryker Fish Date: Fri, 11 Jul 2025 14:14:34 -0600 Subject: [PATCH 2/2] build: fix cuda_arch args --- setup.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 195f71d4..45e24a85 100644 --- a/setup.py +++ b/setup.py @@ -53,9 +53,12 @@ def build_extension(self, ext): "-DPYTHON_EXECUTABLE=" + sys.executable, "-DCMAKE_BUILD_TYPE=" + ("Debug" if self.debug else "Release"), f"-DDOUBLEPRECISION={double_precision}", - f"-DCMAKE_CUDA_ARCHITECTURES={os.environ.get('CMAKE_CUDA_ARCHITECTURES', '')}", ] + cuda_arch = os.environ.get("CMAKE_CUDA_ARCHITECTURES", "") + if cuda_arch: + cmake_args.append(f"-DCMAKE_CUDA_ARCHITECTURES={cuda_arch}") + num_jobs = os.cpu_count() build_args = ["--config", "Release", f"-j{num_jobs}"]