Add compute capability 5.0, 7.5, and 8.0

2025-12-11 16:26:59 +00:00 · 2024-01-20 12:15:50 -08:00
parent 681a914990
commit a447a083f2
5 changed files with 14 additions and 6 deletions
--- a/llm/generate/gen_common.sh
+++ b/llm/generate/gen_common.sh
@@ -39,6 +39,9 @@ init_vars() {
    *)
        ;;
    esac
+    if [ -z "${CMAKE_CUDA_ARCHITECTURES}" ] ; then 
+        CMAKE_CUDA_ARCHITECTURES="50;52;61;70;75;80"
+    fi
 }

 git_module_setup() {