Add GPU env vars to github CI + add device extension to build dir due…

… to concurrent builds
plaidml · Aug 20, 2024 · 74881ce · 74881ce
1 parent 90ba85b
commit 74881ce
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/tpp-llvm.yml b/.github/workflows/tpp-llvm.yml
@@ -26,7 +26,7 @@ jobs:
       - uses: actions/checkout@v4
       - name: LLVM CUDA
         run: |-
-              scripts/buildkite/check_llvm.sh || \
+              GPU=cuda scripts/buildkite/check_llvm.sh || \
               ${{ env.SRUN }} --partition=a100,v100 --time=0:30:00 -- \
               'KIND=RelWithDebInfo COMPILER=clang GPU=cuda \
               ${{ github.workspace }}/scripts/buildkite/build_llvm.sh'
@@ -37,8 +37,7 @@ jobs:
       - uses: actions/checkout@v4
       - name: LLVM Vulkan
         run: |-
-              scripts/buildkite/check_llvm.sh || \
+              GPU=vulkan scripts/buildkite/check_llvm.sh || \
               ${{ env.SRUN }} --partition=spr-all --time=0:30:00 -- \
               'KIND=RelWithDebInfo COMPILER=clang GPU=vulkan \
               ${{ github.workspace }}/scripts/buildkite/build_llvm.sh'
-
diff --git a/scripts/buildkite/build_llvm.sh b/scripts/buildkite/build_llvm.sh
@@ -92,6 +92,7 @@ check_program ${LINKER}
 if [ ! "${LLVM_BUILD_DIR}" ]; then
   LLVM_BUILD_DIR="/scratch/tpp-llvm"
 fi
+LLVM_BUILD_DIR=$(add_device_extensions ${LLVM_BUILD_DIR} ${GPU})
 LLVM_BUILD_DIR=$(realpath ${LLVM_BUILD_DIR})
 LLVM_BUILD_DIR=${LLVM_BUILD_DIR:-build-${COMPILER}}
 mkdir -p ${LLVM_BUILD_DIR}