Install llvm18 packages for ASAN workflows (pytorch#137335)

Follows pytorch#128763 Pull Request resolved: pytorch#137335 Approved by: https://github.com/ezyang
jessejay-ch · Oct 13, 2024 · a90b920 · a90b920
1 parent 4a8e493
commit a90b920
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/.ci/docker/common/install_clang.sh b/.ci/docker/common/install_clang.sh
@@ -20,9 +20,10 @@ if [ -n "$CLANG_VERSION" ]; then
   fi
 
   sudo apt-get update
-  apt-get install -y --no-install-recommends clang-"$CLANG_VERSION" llvm-"$CLANG_VERSION"
-  if [[ $CLANG_VERSION == 18 ]]; then
-    apt-get install -y --no-install-recommends libomp-18-dev
+  if [[ $CLANG_VERSION -ge 18 ]]; then
+    apt-get install -y libomp-${CLANG_VERSION}-dev libclang-rt-${CLANG_VERSION}-dev clang-"$CLANG_VERSION" llvm-"$CLANG_VERSION"
+  else
+    apt-get install -y --no-install-recommends clang-"$CLANG_VERSION" llvm-"$CLANG_VERSION"
   fi
 
   # Install dev version of LLVM.