Merge pull request #473 from nzwulfin/main

Enable GCC Toolet 12 to support AVX VNNI
containers · Nov 20, 2024 · 1529a9b · 1529a9b
2 parents 188a44e + ae139a8
commit 1529a9b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/container-images/scripts/build_llama_and_whisper.sh b/container-images/scripts/build_llama_and_whisper.sh
@@ -29,7 +29,8 @@ dnf_install() {
   elif [ "$containerfile" = "rocm" ]; then
     dnf install -y rocm-dev hipblas-devel rocblas-devel
   elif [ "$containerfile" = "cuda" ]; then
-    dnf install -y "${rpm_list[@]}"
+    dnf install -y "${rpm_list[@]}" gcc-toolset-12
+    source /opt/rh/gcc-toolset-12/enable
   fi
 
   # For Vulkan image, we don't need to install anything extra but rebuild with