From 6beb68993a0c82ae7b2b1566c2bf026b61e6de21 Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Fri, 1 Nov 2024 10:10:46 +0800
Subject: [PATCH] Fix CI

---
 .../workflows/aarch64-linux-gnu-shared.yaml   | 73 +++++++++++++++----
 1 file changed, 60 insertions(+), 13 deletions(-)

diff --git a/.github/workflows/aarch64-linux-gnu-shared.yaml b/.github/workflows/aarch64-linux-gnu-shared.yaml
index 8134b7e8c..1f548e237 100644
--- a/.github/workflows/aarch64-linux-gnu-shared.yaml
+++ b/.github/workflows/aarch64-linux-gnu-shared.yaml
@@ -80,15 +80,24 @@ jobs:
           make -j2
           make install
 
-      - name: cache-toolchain
-        id: cache-toolchain
+      - name: cache-toolchain (CPU)
+        if: matrix.gpu == 'OFF'
+        id: cache-toolchain-cpu
         uses: actions/cache@v4
         with:
           path: toolchain
           key: gcc-linaro-7.5.0-2019.12-x86_64_aarch64-linux-gnu.tar.xz
 
-      - name: Download toolchain
-        if: steps.cache-toolchain.outputs.cache-hit != 'true'
+      - name: cache-toolchain (GPU)
+        if: matrix.gpu == 'ON'
+        id: cache-toolchain-gpu
+        uses: actions/cache@v4
+        with:
+          path: toolchain
+          key: gcc-arm-10.3-2021.07-x86_64-aarch64-none-linux-gnu.tar.xz
+
+      - name: Download toolchain (CPU, gcc 7.5)
+        if: steps.cache-toolchain-cpu.outputs.cache-hit != 'true' && matrix.gpu == 'OFF'
         shell: bash
         run: |
           wget -qq https://huggingface.co/csukuangfj/sherpa-ncnn-toolchains/resolve/main/gcc-linaro-7.5.0-2019.12-x86_64_aarch64-linux-gnu.tar.xz
@@ -96,6 +105,15 @@ jobs:
           mkdir $GITHUB_WORKSPACE/toolchain
           tar xf ./gcc-linaro-7.5.0-2019.12-x86_64_aarch64-linux-gnu.tar.xz --strip-components 1 -C $GITHUB_WORKSPACE/toolchain
 
+      - name: Download toolchain (GPU, gcc 10.3)
+        if: steps.cache-toolchain-gpu.outputs.cache-hit != 'true' && matrix.gpu == 'ON'
+        shell: bash
+        run: |
+          wget -qq https://huggingface.co/csukuangfj/sherpa-ncnn-toolchains/resolve/main/gcc-arm-10.3-2021.07-x86_64-aarch64-none-linux-gnu.tar.xz
+
+          mkdir $GITHUB_WORKSPACE/toolchain
+          tar xf ./gcc-arm-10.3-2021.07-x86_64-aarch64-none-linux-gnu.tar.xz --strip-components 1 -C $GITHUB_WORKSPACE/toolchain
+
       - name: Set environment variable
         if: steps.cache-build-result.outputs.cache-hit != 'true'
         shell: bash
@@ -104,19 +122,31 @@ jobs:
           echo "$GITHUB_WORKSPACE/bin"  >> "$GITHUB_PATH"
           ls -lh "$GITHUB_WORKSPACE/toolchain/bin"
 
-          echo "CC=aarch64-linux-gnu-gcc" >> "$GITHUB_ENV"
-          echo "CXX=aarch64-linux-gnu-g++" >> "$GITHUB_ENV"
+          if [[ ${{ matrix.gpu }} == OFF ]]; then
+            echo "CC=aarch64-linux-gnu-gcc" >> "$GITHUB_ENV"
+            echo "CXX=aarch64-linux-gnu-g++" >> "$GITHUB_ENV"
+          else
+            echo "CC=aarch64-none-linux-gnu-gcc" >> "$GITHUB_ENV"
+            echo "CXX=aarch64-none-linux-gnu-g++" >> "$GITHUB_ENV"
+          fi
 
       - name: Display toolchain info
         shell: bash
         run: |
-          aarch64-linux-gnu-gcc --version
+          if [[ ${{ matrix.gpu }} == OFF ]]; then
+            which aarch64-linux-gnu-gcc
+            aarch64-linux-gnu-gcc --version
+          else
+            which aarch64-none-linux-gnu-gcc
+            aarch64-none-linux-gnu-gcc --version
+          fi
 
       - name: Display qemu-aarch64 -h
         shell: bash
         run: |
           export PATH=$GITHUB_WORKSPACE/qemu-install/bin:$PATH
           export QEMU_LD_PREFIX=$GITHUB_WORKSPACE/toolchain/aarch64-linux-gnu/libc
+          export QEMU_LD_PREFIX=$GITHUB_WORKSPACE/toolchain/aarch64-none-linux-gnu/libc
           qemu-aarch64 -h
 
       - name: build aarch64-linux-gnu
@@ -142,7 +172,11 @@ jobs:
         run: |
           export PATH=$GITHUB_WORKSPACE/toolchain/bin:$PATH
           export PATH=$GITHUB_WORKSPACE/qemu-install/bin:$PATH
-          export QEMU_LD_PREFIX=$GITHUB_WORKSPACE/toolchain/aarch64-linux-gnu/libc
+          if [[ ${{ matrix.gpu }} == OFF ]]; then
+            export QEMU_LD_PREFIX=$GITHUB_WORKSPACE/toolchain/aarch64-linux-gnu/libc
+          else
+            export QEMU_LD_PREFIX=$GITHUB_WORKSPACE/toolchain/aarch64-none-linux-gnu/libc
+          fi
 
           ls -lh ./build-aarch64-linux-gnu/bin
 
@@ -153,11 +187,20 @@ jobs:
       - name: Copy files
         shell: bash
         run: |
-          aarch64-linux-gnu-strip --version
+          if [[ ${{ matrix.gpu }} == OFF ]]; then
+            aarch64-linux-gnu-strip --version
+          else
+            aarch64-none-linux-gnu-strip --version
+          fi
 
           SHERPA_ONNX_VERSION=v$(grep "SHERPA_ONNX_VERSION" ./CMakeLists.txt  | cut -d " " -f 2  | cut -d '"' -f 2)
 
           dst=sherpa-onnx-${SHERPA_ONNX_VERSION}-linux-aarch64-shared
+          if [[ ${{ matrix.gpu }} == OFF ]]; then
+            dst=${dst}-cpu
+          else
+            dst=${dst}-gpu
+          fi
           mkdir $dst
 
           cp -a build-aarch64-linux-gnu/install/bin $dst/
@@ -168,7 +211,11 @@ jobs:
 
           ls -lh $dst/bin/
           echo "strip"
-          aarch64-linux-gnu-strip $dst/bin/*
+          if [[ ${{ matrix.gpu }} == OFF ]]; then
+            aarch64-linux-gnu-strip $dst/bin/*
+          else
+            aarch64-none-linux-gnu-strip $dst/bin/*
+          fi
 
           tree $dst
 
@@ -176,8 +223,8 @@ jobs:
 
       - uses: actions/upload-artifact@v4
         with:
-          name: sherpa-onnx-linux-aarch64-shared
-          path: sherpa-onnx-*linux-aarch64-shared.tar.bz2
+          name: sherpa-onnx-linux-aarch64-shared-gpu-${{ matrix.gpu }}
+          path: sherpa-onnx-*linux-aarch64-shared*.tar.bz2
 
       # https://huggingface.co/docs/hub/spaces-github-actions
       - name: Publish to huggingface
@@ -200,7 +247,7 @@ jobs:
             cd huggingface
             mkdir -p aarch64
 
-            cp -v ../sherpa-onnx-*-shared.tar.bz2 ./aarch64
+            cp -v ../sherpa-onnx-*-shared*.tar.bz2 ./aarch64
 
             git status
             git lfs track "*.bz2"