build-run-mlperf-tiny.yml

name: Build and run MLPerf Tiny Networks

on:
  push:
    branches:
      - main
  pull_request:
  workflow_dispatch:
    inputs:
      container-tag:
        description: 'Container tag'
        required: true
        default: 'main'

jobs:
  build-and-run-networks:
    runs-on: ubuntu-latest
    container: 
      image: ghcr.io/kuleuven-micas/snax-mlir:${{ github.event.inputs.container-tag || 'main' }}
    steps:
      - uses: actions/checkout@v3
      - name: Install snax compiler
        run: |
          /opt/python3.11/bin/python3 -m pip install -e .
      - name: Reinstall pip modules from requirements
        run: |
          /opt/python3.11/bin/python3 -m pip install -r requirements.txt
      - name: Anomaly detection
        run: |
          export PATH=/opt/python3.11/bin:$PATH
          make ad01_int8.preproc.mlir
          xdsl-opt ad01_int8.preproc.mlir
        working-directory: kernels/mlperf_tiny_ad01
      - name: Image classfication
        run: |
          export PATH=/opt/python3.11/bin:$PATH
          make pretrainedResnet_quant.preproc.mlir
        working-directory: kernels/mlperf_tiny_ic
      - name: Visual Wake Words
        run: |
          export PATH=/opt/python3.11/bin:$PATH
          make vww_96_int8.preproc.mlir
        working-directory: kernels/mlperf_tiny_vww
      - name: Keyword spotting
        run: |
          export PATH=/opt/python3.11/bin:$PATH
          make kws_ref_model.preproc.mlir
        working-directory: kernels/mlperf_tiny_kws