From 973de5380bc616bfb7e50469cfd4959cee91dbb2 Mon Sep 17 00:00:00 2001
From: Christian Heitman <cheitman@quarkslab.com>
Date: Wed, 25 Oct 2023 15:32:23 -0300
Subject: [PATCH] Update workflow python

---
 .github/workflows/python.yml            | 11 ++++++++++-
 src/scripts/docker/build-wheel-linux.sh |  3 ++-
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/python.yml b/.github/workflows/python.yml
index 2368bbaa2..4972006d8 100644
--- a/.github/workflows/python.yml
+++ b/.github/workflows/python.yml
@@ -15,11 +15,20 @@ jobs:
       - name: Create Docker image
         run: bash ./src/scripts/docker/build-docker-image.sh
 
+      - name: Install LLVM and Clang
+        uses: KyleMayes/install-llvm-action@v1
+        with:
+          version: "16.0"
+          directory: ${{ runner.temp }}/llvm
+
       - name: Build Wheel packages
         run: |
-         docker run \
+          echo "LLVM PATH: " ${{env.LLVM_PATH}}
+          ls ${{env.LLVM_PATH}}
+          docker run \
                 --rm \
                 --volume $GITHUB_WORKSPACE:/src \
+                --volume ${{env.LLVM_PATH}}:/llvm \
                 build-triton-linux-x86_64 bash /src/src/scripts/docker/build-wheel-linux.sh
 
       - name: Upload Wheel packages (Python 3.8)
diff --git a/src/scripts/docker/build-wheel-linux.sh b/src/scripts/docker/build-wheel-linux.sh
index 9855c2a01..be34885cd 100755
--- a/src/scripts/docker/build-wheel-linux.sh
+++ b/src/scripts/docker/build-wheel-linux.sh
@@ -15,6 +15,7 @@ set -e
 # set -x  # Debugging
 
 DEPENDENCIES_DIR=/tmp/triton-dependencies
+LLVM_DIR=/llvm
 SOURCE_DIR=/src
 WHEEL_DIR=$SOURCE_DIR/wheelhouse
 
@@ -28,7 +29,7 @@ export BITWUZLA_INTERFACE=On
 export BITWUZLA_INCLUDE_DIRS=$DEPENDENCIES_DIR/bitwuzla/install/include
 export BITWUZLA_LIBRARIES=$DEPENDENCIES_DIR/bitwuzla/install/lib64/libbitwuzla.so
 export LLVM_INTERFACE=ON
-export CMAKE_PREFIX_PATH=$($DEPENDENCIES_DIR/clang+llvm-12.0.1-x86_64-linux-gnu-ubuntu-/bin/llvm-config --prefix)
+export CMAKE_PREFIX_PATH=$LLVM_DIR
 
 # Build Triton Python wheel package for Python 3.8.
 echo "[+] Build Triton wheel package for Python 3.8"