From aa9d96f8aff50e54e65cfa06eb3957df617369b4 Mon Sep 17 00:00:00 2001 From: wsh Date: Thu, 31 Jul 2025 16:11:33 +0800 Subject: [PATCH 1/2] support rtx5090 sm_120 --- CMakeLists.txt | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index f0ca595..77a13a9 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -87,7 +87,7 @@ if(GPU_RUNTIME STREQUAL "CUDA") endif() if(NOT OPENSPLAT_MAX_CUDA_COMPATIBILITY) if(NOT CMAKE_CUDA_ARCHITECTURES) - SET(CMAKE_CUDA_ARCHITECTURES 70;75;80) + SET(CMAKE_CUDA_ARCHITECTURES 70;75;80;120) endif() else() # Build for maximum compatibility @@ -119,7 +119,8 @@ if(GPU_RUNTIME STREQUAL "CUDA") endif() # Set torch cuda architecture list set(TORCH_CUDA_ARCH_LIST ${CMAKE_CUDA_ARCHITECTURES}) - list(TRANSFORM TORCH_CUDA_ARCH_LIST REPLACE "([0-9])([0-9])" "\\1.\\2") + list(TRANSFORM TORCH_CUDA_ARCH_LIST REPLACE "^([0-9])([0-9])$" "\\1.\\2") + list(TRANSFORM TORCH_CUDA_ARCH_LIST REPLACE "^([0-9][0-9])([0-9])$" "\\1.\\2") string(REPLACE ";" " " TORCH_CUDA_ARCH_LIST "${TORCH_CUDA_ARCH_LIST}") message(STATUS "** Updated TORCH_CUDA_ARCH_LIST to ${TORCH_CUDA_ARCH_LIST} **") endif() From 0423f87e6f86d1cbb689c61b203a1963c5aa9c61 Mon Sep 17 00:00:00 2001 From: wsh Date: Fri, 1 Aug 2025 11:09:11 +0800 Subject: [PATCH 2/2] =?UTF-8?q?Partially=20revert=20previous=20changes=20a?= =?UTF-8?q?nd=20correct=20the=20regular=20expression=20that=20mistakenly?= =?UTF-8?q?=20converted=20=E2=80=9C120=E2=80=9D=20to=20=E2=80=9C1.20?= =?UTF-8?q?=E2=80=9D.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 77a13a9..625f062 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -87,7 +87,7 @@ if(GPU_RUNTIME STREQUAL "CUDA") endif() if(NOT OPENSPLAT_MAX_CUDA_COMPATIBILITY) if(NOT CMAKE_CUDA_ARCHITECTURES) - SET(CMAKE_CUDA_ARCHITECTURES 70;75;80;120) + SET(CMAKE_CUDA_ARCHITECTURES 70;75;80) endif() else() # Build for maximum compatibility