Added support for NVIDIA GeForce RTX 4090

asifpatankar · asifpatankar · commit b6c12748f50f · 2024-07-11T15:43:30.000+09:00
Signed-off-by: asifpatankar &lt;asifpatankar@roketmail.com&gt;
diff --git 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()

Original file line number	Diff line number	Diff line change
`@@ -154,6 +154,9 @@ else()`
`154`	`154`	`61`
`155`	`155`	`70`
`156`	`156`	`75`
	`157`	`+ 80`
	`158`	`+ 86`
	`159`	`+ 89 # Adding support for NVIDIA GeForce RTX 4090`
`157`	`160`	`)`
`158`	`161`
`159`	`162`	`string(REGEX MATCH "aarch64" IS_ARM "${TRT_PLATFORM_ID}")`
`@@ -169,6 +172,9 @@ else()`
`169`	`172`	`if (CUDA_VERSION VERSION_GREATER_EQUAL 11.1)`
`170`	`173`	`list(APPEND GPU_ARCHS 86)`
`171`	`174`	`endif()`
	`175`	`+ if (CUDA_VERSION VERSION_GREATER_EQUAL 12.1)`
	`176`	`+ list(APPEND GPU_ARCHS 89)`
	`177`	`+ endif()`
`172`	`178`
`173`	`179`	`message(STATUS "GPU_ARCHS is not defined. Generating CUDA code for default SMs: ${GPU_ARCHS}")`
`174`	`180`	`endif()`