diff --git a/.github/workflows/_build_linux.yml b/.github/workflows/_build_linux.yml
index 5c609e5..e91163c 100644
--- a/.github/workflows/_build_linux.yml
+++ b/.github/workflows/_build_linux.yml
@@ -104,6 +104,7 @@ jobs:
 
       - name: Apply auditwheel repair
         continue-on-error: true
+        id: auditwheel_repair
         run: |
           auditwheel show ${{ steps.build_wheels.outputs.WHEEL_PATH }}
           auditwheel repair \
@@ -116,7 +117,7 @@ jobs:
         continue-on-error: true
         run: |
           pip uninstall -y flash-attn > /dev/null 2>&1
-          pip install --no-cache-dir ${{ steps.build_wheels.outputs.WHEEL_PATH_MANYLINUX }}
+          pip install --no-cache-dir ${{ steps.auditwheel_repair.outputs.WHEEL_PATH_MANYLINUX }}
           python -c "import flash_attn; print(flash_attn.__version__)"
 
       - name: Upload manylinux wheel
@@ -124,7 +125,7 @@ jobs:
         env:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         run: |
-          wheel_path_manylinux="${{ steps.build_wheels.outputs.WHEEL_PATH_MANYLINUX }}"
+          wheel_path_manylinux="${{ steps.auditwheel_repair.outputs.WHEEL_PATH_MANYLINUX }}"
           if [ ! -f "$wheel_path_manylinux" ]; then
             echo "Error: Wheel file not found at $wheel_path_manylinux"
             exit 1
diff --git a/.github/workflows/_build_linux_arm_self_host.yml b/.github/workflows/_build_linux_arm_self_host.yml
index 3985f68..2eb11b1 100644
--- a/.github/workflows/_build_linux_arm_self_host.yml
+++ b/.github/workflows/_build_linux_arm_self_host.yml
@@ -73,7 +73,8 @@ jobs:
             clang \
             ninja-build \
             keyboard-configuration \
-            time
+            time \
+            patchelf
 
       - name: Install gh
         shell: bash
@@ -146,6 +147,7 @@ jobs:
 
       - name: Apply auditwheel repair
         continue-on-error: true
+        id: auditwheel_repair
         run: |
           auditwheel show ${{ steps.build_wheels.outputs.WHEEL_PATH }}
           auditwheel repair \
@@ -158,7 +160,7 @@ jobs:
         continue-on-error: true
         run: |
           pip uninstall -y flash-attn > /dev/null 2>&1
-          pip install --no-cache-dir ${{ steps.build_wheels.outputs.WHEEL_PATH_MANYLINUX }}
+          pip install --no-cache-dir ${{ steps.auditwheel_repair.outputs.WHEEL_PATH_MANYLINUX }}
           python -c "import flash_attn; print(flash_attn.__version__)"
 
       - name: Upload manylinux wheel
@@ -166,7 +168,7 @@ jobs:
         env:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         run: |
-          wheel_path_manylinux="${{ steps.build_wheels.outputs.WHEEL_PATH_MANYLINUX }}"
+          wheel_path_manylinux="${{ steps.auditwheel_repair.outputs.WHEEL_PATH_MANYLINUX }}"
           if [ ! -f "$wheel_path_manylinux" ]; then
             echo "Error: Wheel file not found at $wheel_path_manylinux"
             exit 1
diff --git a/.github/workflows/_build_linux_self_host.yml b/.github/workflows/_build_linux_self_host.yml
index 5129dee..c5c25a3 100644
--- a/.github/workflows/_build_linux_self_host.yml
+++ b/.github/workflows/_build_linux_self_host.yml
@@ -152,6 +152,7 @@ jobs:
 
       - name: Apply auditwheel repair
         continue-on-error: true
+        id: auditwheel_repair
         run: |
           auditwheel show ${{ steps.build_wheels.outputs.WHEEL_PATH }}
           auditwheel repair \
@@ -164,7 +165,7 @@ jobs:
         continue-on-error: true
         run: |
           pip uninstall -y flash-attn > /dev/null 2>&1
-          pip install --no-cache-dir ${{ steps.build_wheels.outputs.WHEEL_PATH_MANYLINUX }}
+          pip install --no-cache-dir ${{ steps.auditwheel_repair.outputs.WHEEL_PATH_MANYLINUX }}
           python -c "import flash_attn; print(flash_attn.__version__)"
 
       - name: Upload manylinux wheel
@@ -172,7 +173,7 @@ jobs:
         env:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         run: |
-          wheel_path_manylinux="${{ steps.build_wheels.outputs.WHEEL_PATH_MANYLINUX }}"
+          wheel_path_manylinux="${{ steps.auditwheel_repair.outputs.WHEEL_PATH_MANYLINUX }}"
           if [ ! -f "$wheel_path_manylinux" ]; then
             echo "Error: Wheel file not found at $wheel_path_manylinux"
             exit 1
@@ -303,6 +304,7 @@ jobs:
 
       - name: Apply auditwheel repair
         continue-on-error: true
+        id: auditwheel_repair
         run: |
           auditwheel show ${{ steps.build_wheels.outputs.WHEEL_PATH }}
           auditwheel repair \
@@ -315,7 +317,7 @@ jobs:
         continue-on-error: true
         run: |
           pip uninstall -y flash-attn > /dev/null 2>&1
-          pip install --no-cache-dir ${{ steps.build_wheels.outputs.WHEEL_PATH_MANYLINUX }}
+          pip install --no-cache-dir ${{ steps.auditwheel_repair.outputs.WHEEL_PATH_MANYLINUX }}
           python -c "import flash_attn; print(flash_attn.__version__)"
 
       - name: Upload manylinux wheel
@@ -323,7 +325,7 @@ jobs:
         env:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         run: |
-          wheel_path_manylinux="${{ steps.build_wheels.outputs.WHEEL_PATH_MANYLINUX }}"
+          wheel_path_manylinux="${{ steps.auditwheel_repair.outputs.WHEEL_PATH_MANYLINUX }}"
           if [ ! -f "$wheel_path_manylinux" ]; then
             echo "Error: Wheel file not found at $wheel_path_manylinux"
             exit 1
diff --git a/.github/workflows/_build_manylinux_self_host.yml b/.github/workflows/_build_manylinux_self_host.yml
index b926947..dab5e51 100644
--- a/.github/workflows/_build_manylinux_self_host.yml
+++ b/.github/workflows/_build_manylinux_self_host.yml
@@ -142,6 +142,8 @@ jobs:
 
       - name: Apply auditwheel repair
         continue-on-error: true
+        id: auditwheel_repair
+        shell: bash
         run: |
           auditwheel show ${{ steps.build_wheels.outputs.WHEEL_PATH }}
           auditwheel repair \
@@ -154,7 +156,7 @@ jobs:
         continue-on-error: true
         run: |
           pip uninstall -y flash-attn > /dev/null 2>&1
-          pip install --no-cache-dir ${{ steps.build_wheels.outputs.WHEEL_PATH_MANYLINUX }}
+          pip install --no-cache-dir ${{ steps.auditwheel_repair.outputs.WHEEL_PATH_MANYLINUX }}
           python -c "import flash_attn; print(flash_attn.__version__)"
 
       - name: Upload manylinux wheel
@@ -162,7 +164,7 @@ jobs:
         env:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         run: |
-          wheel_path_manylinux="${{ steps.build_wheels.outputs.WHEEL_PATH_MANYLINUX }}"
+          wheel_path_manylinux="${{ steps.auditwheel_repair.outputs.WHEEL_PATH_MANYLINUX }}"
           if [ ! -f "$wheel_path_manylinux" ]; then
             echo "Error: Wheel file not found at $wheel_path_manylinux"
             exit 1