From b6c12748f50f78496a13af5345c523afb6d990ab Mon Sep 17 00:00:00 2001 From: asifpatankar Date: Thu, 11 Jul 2024 15:36:06 +0900 Subject: [PATCH] Added support for NVIDIA GeForce RTX 4090 Signed-off-by: asifpatankar --- CMakeLists.txt | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/CMakeLists.txt b/CMakeLists.txt index 66f4201b..1fe8b381 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -154,6 +154,9 @@ else() 61 70 75 + 80 + 86 + 89 # Adding support for NVIDIA GeForce RTX 4090 ) string(REGEX MATCH "aarch64" IS_ARM "${TRT_PLATFORM_ID}") @@ -169,6 +172,9 @@ else() if (CUDA_VERSION VERSION_GREATER_EQUAL 11.1) list(APPEND GPU_ARCHS 86) endif() + if (CUDA_VERSION VERSION_GREATER_EQUAL 12.1) + list(APPEND GPU_ARCHS 89) + endif() message(STATUS "GPU_ARCHS is not defined. Generating CUDA code for default SMs: ${GPU_ARCHS}") endif()