tflm.yml

name: '🔥 Build TFLM'

on:
  push:
    tags:
      - 'v*.*.*'
    branches:    
      - 'main'
    paths:
      - 'src/**'
      - '.github/workflows/*.yml'
      - '.github/workflows/*.json'
      - '!**/README.md'
      - '!**.rst'

  pull_request:
    branches:    
      - 'main'
    paths:
      - 'src/**'
      - '.github/workflows/*.yml'
      - '.github/workflows/*.json'
      - '!**/README.md'
      - '!**.rst'

jobs:
  build-tflm:
    runs-on: ubuntu-20.04

    strategy:
      matrix:
        target:
          - {name: cortex-m0plus, arch: cortex-m0plus, kernel: cmsis_nn, coproc: '', args: ''}
          - {name: cortex-m4+fp, arch: cortex-m4+fp, kernel: cmsis_nn, coproc: '', args: ''}
          - {name: cortex-m7+fp, arch: cortex-m7+fp, kernel: cmsis_nn, coproc: '', args: ''}
          - {name: cortex-m55, arch: cortex-m55, kernel: cmsis_nn, coproc: '', args: ''}
          - {name: cortex-m55-u55, arch: cortex-m55, kernel: ethos_u, coproc: ethos_u, args: "ETHOSU_ARCH=u55"}
        build: [release, debug]
      fail-fast: true

    steps:
    - name: '⏳ Checkout main'
      uses: actions/checkout@v4
      with:
        submodules: false

    - name: '⏳ Checkout TFLM'
      uses: actions/checkout@v4
      with:
        submodules: false
        path: tflite-micro
        repository: tensorflow/tflite-micro
        ref: 8c458fc48ee972ea4fc56ddc08849823b9af7ea8

    - name: '♻ Caching dependencies'
      uses: actions/cache@v4.0.2
      id: cache
      with:
        path: ~/cache/gcc
        key: 'arm-gnu-toolchain-13.2.rel1'

    - name: '🛠 Install toolchain '
      if: steps.cache.outputs.cache-hit != 'true'
      run: source tools/ci.sh && ci_install_arm_gcc

    - name: '🛠 Install dependencies'
      run: |
        python -m pip install --upgrade pip numpy==1.24 Pillow==10.3

    - name: '🏗 Build TFLM'
      run: |
        source tools/ci.sh
        ci_build_target ${{ matrix.target.name }} \
                        ${{ matrix.target.arch }} \
                        "${{ matrix.target.kernel }}" \
                        "${{ matrix.target.coproc }}" \
                        "${{ matrix.target.args }}" \
                        ${{ matrix.build }}

    - name: '⬆ Upload library'
      uses: actions/upload-artifact@v4
      with:
        name: ${{ matrix.target.name }}-${{ matrix.build }}
        path: libs
        retention-days: 1
        if-no-files-found: error

    - name: '🏗 Build headers'
      if: matrix.build == 'release'
      run: |
        source tools/ci.sh
        ci_build_headers ${{ matrix.target.name }} \
                        ${{ matrix.target.arch }} \
                        "${{ matrix.target.kernel }}" \
                        "${{ matrix.target.coproc }}" \
                        "${{ matrix.target.args }}" \
                        release

    - name: '⬆ Upload headers'
      if: matrix.build == 'release'
      uses: actions/upload-artifact@v4
      with:
        name: ${{ matrix.target.name }}-headers
        path: ${{ matrix.target.name }}-headers
        retention-days: 1
        if-no-files-found: error

  push-artifacts:
    runs-on: ubuntu-20.04
    needs: [build-tflm]
#    if: github.event_name == 'push'
    permissions:
      contents: write
      pull-requests: write

    steps:
    - name: '⏳ Checkout repository'
      uses: actions/checkout@v4
      with:
        submodules: false

    - name: '⬇ Download artifacts'
      uses: actions/download-artifact@v4

    - name: '🔄 Update headers'
      run: |
        rsync -av --delete --include='*/' --include='LICENSE' --include='*.h' --exclude='*' --delete *-headers/ include
        rm -fr *-headers

    - name: '🔀 Create Pull Request'
      uses: peter-evans/create-pull-request@v5
      with:
        base: main
        branch: update_tflm
        title: Update tflm libraries and headers.
        commit-message: Update tflm libraries and headers.
      env:
        GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}