test-mlc-based-submission-generation.yml

name: MLC based Submission Generation

on:
  pull_request:
    branches: [ "main", "dev" ]
    paths:
      - '.github/workflows/test-mlc-based-submission-generation.yml'
      - '**'  
      - '!**.md'
jobs:
  submission_generation:
    runs-on: ${{ matrix.os }}
    strategy:
      fail-fast: false
      matrix:
        os: [ubuntu-latest, windows-latest, macos-latest]
        python-version: [ "3.12" ]
        division: ["closed", "open", "closed-open"]
        category: ["datacenter", "edge"]
        case: ["closed", "closed-no-compliance", "closed-power", "closed-failed-power-logs", "case-1", "case-2", "case-3", "case-5", "case-6", "case-7", "case-8"]
        action: ["run", "docker"]
        exclude:
          - os: macos-latest
          - os: windows-latest
          - category: "edge"
          - case: case-1
            division: closed
          - case: case-1
            division: closed-open
          - case: case-2
            division: closed
          - case: case-2
            division: closed-open
          - case: case-5
            division: closed
          - case: case-5
            division: closed-open
          - case: case-6
            division: closed
          - case: case-6
            division: closed-open
            
    steps:
    - uses: actions/checkout@v4
    - name: Set up Python ${{ matrix.python-version }}
      uses: actions/setup-python@v3
      with:
        python-version: ${{ matrix.python-version }}
    - name: Install mlcflow
      run: |
        pip install mlcflow
    - name: Pull MLOps repository
      run: |
        mlc pull repo ${{ github.event.pull_request.head.repo.html_url }} --branch=${{ github.event.pull_request.head.ref }}
    - name: Pull repo where test cases are uploaded
      run: |
        git clone -b submission-generation-tests https://github.com/mlcommons/inference.git submission_generation_tests
    
    - name: Run Submission Generation - ${{ matrix.case }} ${{ matrix.action }} ${{ matrix.category }} ${{ matrix.division }} 
      continue-on-error: true
      run: |
        if [ "${{ matrix.case }}" == "case-3" ]; then
          description="Submission generation (model_mapping.json not present but model name matches with official one)"
        elif [ "${{ matrix.case }}" == "case-7" ]; then
          description="Submission generation (sut_info.json incomplete, SUT folder name in required format)"
        elif [ "${{ matrix.case }}" == "case-8" ]; then
          extra_run_args=" --category=datacenter"
          description="Submission generation (system_meta.json not found in results folder)"
        elif [ "${{ matrix.case }}" == "closed" ]; then
          extra_run_args=" --env.MLC_MLPERF_SUBMISSION_CHECKER_EXTRA_ARGS="--skip-extra-accuracy-files-check""
          description="Test submission - contains closed edge and datacenter"
        elif [ "${{ matrix.case }}" == "closed-no-compliance" ]; then
          extra_run_args=" --env.MLC_MLPERF_SUBMISSION_CHECKER_EXTRA_ARGS="--skip-extra-accuracy-files-check""
          description="Test submission - contains closed edge and datacenter with no compliance tests"
        elif [ "${{ matrix.case }}" == "closed-power" ]; then
          extra_run_args=" --env.MLC_MLPERF_SUBMISSION_CHECKER_EXTRA_ARGS="--skip-extra-accuracy-files-check""
          description="Test submission - contains closed-power edge and datacenter results"
        elif [ "${{ matrix.case }}" == "closed-failed-power-logs" ]; then
          extra_run_args=" --env.MLC_MLPERF_SUBMISSION_CHECKER_EXTRA_ARGS="--skip-extra-accuracy-files-check""
          description="Test submission - contains closed-power edge and datacenter results with failed power logs"
        fi
        # Dynamically set the log group to simulate a dynamic step name
        echo "::group::$description"
        mlc ${{ matrix.action }} script --tags=generate,inference,submission  --version=v4.1 --clean --preprocess_submission=yes --results_dir=$PWD/submission_generation_tests/${{ matrix.case }}/ --run-checker --submitter=MLCommons --tar=yes --env.MLC_TAR_OUTFILE=submission.tar.gz --division=${{ matrix.division }} --env.MLC_DETERMINE_MEMORY_CONFIGURATION=yes --quiet $extra_run_args
        exit_status=$?
        echo "Exit status for the job ${description} ${exit_status}"
        if [[ "${{ matrix.case }}" == "case-5" || "${{ matrix.case }}" == "case-6" ]]; then
          # For cases 5 and 6, exit status should be 0 if cm command fails, 1 if it succeeds
          if [[ ${exit_status} -ne 0 ]]; then
            echo "STEP_FAILED=false" >> $GITHUB_ENV
          else
            echo "STEP_FAILED=true" >> $GITHUB_ENV
          fi
        else
          # For other cases, exit with the original status
          if [[ ${exit_status} -eq 0 ]]; then
            echo "STEP_FAILED=false" >> $GITHUB_ENV
          else
            echo "STEP_FAILED=true" >> $GITHUB_ENV
          fi
        fi
        echo "::endgroup::"
    - name: Fail if Step Failed
      if: env.STEP_FAILED == 'true'
      continue-on-error: false
      run: |
        echo "Manually failing the workflow because the step failed."
        exit 1