From 3968bf43f687767898597c9f284033c508b42e1b Mon Sep 17 00:00:00 2001 From: sfc-gh-zhwang Date: Thu, 14 Sep 2023 21:18:39 -0700 Subject: [PATCH 1/2] commit --- CMakeLists.txt | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index a164ef827..c6169f0e2 100755 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -167,16 +167,17 @@ if(NOT (FIND_SM STREQUAL True)) -gencode=arch=compute_75,code=\\\"sm_75,compute_75\\\" \ -gencode=arch=compute_80,code=\\\"sm_80,compute_80\\\" \ -gencode=arch=compute_86,code=\\\"sm_86,compute_86\\\" \ + -gencode=arch=compute_90,code=\\\"sm_86,compute_90\\\" \ ") # -rdc=true") set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -DWMMA") set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DWMMA") set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -DWMMA") if(BUILD_PYT) - set(ENV{TORCH_CUDA_ARCH_LIST} "7.0;7.5;8.0;8.6") + set(ENV{TORCH_CUDA_ARCH_LIST} "7.0;7.5;8.0;8.6;9.0") endif() - set(CMAKE_CUDA_ARCHITECTURES 70 75 80 86) - message("-- Assign GPU architecture (sm=70,75,80,86)") + set(CMAKE_CUDA_ARCHITECTURES 70 75 80 86 90) + message("-- Assign GPU architecture (sm=70,75,80,86,90)") endif() if(BUILD_PYT) From 6c608049c92a51902f8e6b38ff3aca19ffc6bed4 Mon Sep 17 00:00:00 2001 From: sfc-gh-zhwang Date: Thu, 14 Sep 2023 21:33:25 -0700 Subject: [PATCH 2/2] commit --- CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index c6169f0e2..ca82997c8 100755 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -167,7 +167,7 @@ if(NOT (FIND_SM STREQUAL True)) -gencode=arch=compute_75,code=\\\"sm_75,compute_75\\\" \ -gencode=arch=compute_80,code=\\\"sm_80,compute_80\\\" \ -gencode=arch=compute_86,code=\\\"sm_86,compute_86\\\" \ - -gencode=arch=compute_90,code=\\\"sm_86,compute_90\\\" \ + -gencode=arch=compute_90,code=\\\"sm_90,compute_90\\\" \ ") # -rdc=true") set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -DWMMA")