finn_integration.yml

name: Test Brevitas-FINN integration

on:
  push:
    branches: [ master, dev ]
  pull_request:
    types:
      - review_requested

jobs:
  build:
    runs-on: ${{ matrix.platform }}
    strategy:
      fail-fast: false


      matrix:
        python_version: ['3.8', '3.9']
        pytorch_version: ['1.9.1', '1.10.1', '1.11.0', '1.12.1', '1.13.0', '2.0.1', '2.1.0']
        platform: ['windows-latest', 'ubuntu-latest']


        exclude:
          - pytorch_version: '1.9.1'
            platform: 'macos-latest'



    if: ${{ !github.event.pull_request.draft }}
    steps:

    - name: Checkout repo
      uses: actions/checkout@v3

    - name: Set up Python
      uses: actions/setup-python@v2
      with:
       python-version: ${{ matrix.python_version }}

    - name: Install Nox dependencies
      shell: bash
      run: pip install -r requirements/requirements-nox.txt

    - name: Install update
      shell: bash
      run: sudo apt-get update
      if: startsWith(runner.os, 'Linux') == true

    - name: Install libsndfile and libgomp1 on Ubuntu
      shell: bash
      run: sudo apt-get install -y libsndfile-dev libgomp1
      if: startsWith(runner.os, 'Linux') == true

    - name: Install libomp on macOS
      shell: bash
      run: brew install libomp
      if: startsWith(runner.os, 'macOS') == true

    - name: Install protobuf on Ubuntu
      shell: bash
      run: sudo apt-get install protobuf-compiler libprotoc-dev
      if: startsWith(runner.os, 'Linux') == true

    - name: Run Nox session for Brevitas-FINN integration
      shell: bash
      run: nox -v -s tests_brevitas_finn_integration-${{ matrix.python_version }}\(\pytorch_${{ matrix.pytorch_version }}\)