name: rknn

on:
  push:
    branches:
      - master
      - ci-rknn

  pull_request:
    branches:
      - master

  workflow_dispatch:

concurrency:
  group: rknn-${{ github.ref }}
  cancel-in-progress: true

jobs:
  rknn:
    runs-on: ubuntu-latest
    strategy:
      fail-fast: false
      matrix:
        os: [ubuntu-latest]
        python-version: ["3.10"]
    steps:
      - uses: actions/checkout@v4
        with:
          fetch-depth: 0

      - name: Setup Python
        uses: actions/setup-python@v5
        with:
          python-version: ${{ matrix.python-version }}

      - name: Install rknn
        shell: bash
        run: |
          ls
          curl -SL -O https://huggingface.co/csukuangfj/rknn-toolkit2/resolve/main/rknn_toolkit2-2.3.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
          pip install ./*.whl "numpy<=1.26.4"
          pip list | grep rknn
          echo "---"
          pip list
          echo "---"

      - name: Run
        shell: bash
        run: |
          ls
          python3 -c "import sys; from rknn.api import RKNN; print(dir(RKNN))"

          git clone --depth 1 https://github.com/airockchip/rknn_model_zoo/
          cd rknn_model_zoo/examples/zipformer
          cd model
          ls -lh
          chmod +x ./download_model.sh
          ./download_model.sh
          ls -lh
          cd ..

          cd python
          python3 ./convert.py ../model/encoder-epoch-99-avg-1.onnx rk3588

          ls -lh ../model

          python3 convert.py ../model/decoder-epoch-99-avg-1.onnx rk3588

          ls -lh ../model

          python3 convert.py ../model/joiner-epoch-99-avg-1.onnx rk3588

          ls -lh ../model

      - uses: actions/upload-artifact@v4
        with:
          name: rknn-files
          path: ./rknn_model_zoo/examples/zipformer/model/*.rknn

      - uses: actions/upload-artifact@v4
        with:
          name: onnx-files
          path: ./rknn_model_zoo/examples/zipformer/model/*.onnx