Skip to content

Add support for tflite models from MLPerf Tiny (#59) #4

Add support for tflite models from MLPerf Tiny (#59)

Add support for tflite models from MLPerf Tiny (#59) #4

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