From d8b5a588989882508a1259095d5e5b8ae6f74d4f Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Wed, 28 May 2025 17:39:55 +0800
Subject: [PATCH] repair rknn wheels (#2257)

---
 .../workflows/build-wheels-aarch64-rknn.yaml  | 25 ++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/build-wheels-aarch64-rknn.yaml b/.github/workflows/build-wheels-aarch64-rknn.yaml
index ab872e9b..71edea74 100644
--- a/.github/workflows/build-wheels-aarch64-rknn.yaml
+++ b/.github/workflows/build-wheels-aarch64-rknn.yaml
@@ -25,6 +25,11 @@ jobs:
         with:
           fetch-depth: 0
 
+      - name: Setup Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+
       - name: Download rknn-toolkit2
         shell: bash
         run: |
@@ -109,10 +114,28 @@ jobs:
               export SHERPA_ONNX_CMAKE_ARGS="-DSHERPA_ONNX_ENABLE_RKNN=ON"
               python3 setup.py bdist_wheel
 
+              mv dist wheelhouse
+
       - name: Display results
         shell: bash
         run: |
-          ls -lh dist
+          ls -lh wheelhouse
+
+      - name: Fix wheel name
+        shell: bash
+        run: |
+          python3 -m pip install auditwheel
+
+          auditwheel show ./wheelhouse/*.whl
+
+          auditwheel repair --help
+
+          auditwheel --verbose repair --plat manylinux_2_27_aarch64 \
+            --exclude librknnrt.so \
+            --exclude libasound.so.2 \
+            -w ./dist ./wheelhouse/*.whl
+
+          ls -lh dist/*.whl
 
       - name: Show glibc versions
         shell: bash