Add LLVM to the environment

mdboom · mdboom · commit 598302b7b4df · 2025-06-27T13:47:19.000-04:00
diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml
@@ -31,6 +31,10 @@ on:
         description: 'Make flags for CPython build'
         required: false
         default: '-j'
+      llvm:
+        description: 'LLVM version to use'
+        required: false
+        default: '19'
 
 jobs:
   benchmark:
@@ -61,8 +65,11 @@ jobs:
         # Install CPython dependencies using their script
         cd cpython
         sudo .github/workflows/posix-deps-apt.sh
-        
-        # Install Memray dependencies  
+
+        # Install JIT dependencies
+        sudo bash -c "$(wget -O - https://apt.llvm.org/llvm.sh)" ./llvm.sh ${{ inputs.llvm }}
+
+        # Install Memray dependencies
         sudo apt-get install -y \
           python3-dev \
           libdebuginfod-dev \
@@ -73,6 +80,8 @@ jobs:
       env:
         MEMORY_TRACKER_TOKEN: ${{ secrets.MEMORY_TRACKER_TOKEN }}
       run: |
+        export PATH="$(llvm-config-${{ inputs.llvm }} --bindir):$PATH"
+
         # Build command with conditional flags
         CMD="memory-tracker benchmark '${{ github.event.inputs.commit_range }}'"
         CMD="$CMD --repo-path ./cpython"
