Bump scheduled_unet_compstat_cpu golden_binary_size to 2000000 (#179) #354
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # Copyright 2025 The IREE Authors | |
| # | |
| # Licensed under the Apache License v2.0 with LLVM Exceptions. | |
| # See https://llvm.org/LICENSE.txt for license information. | |
| # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception | |
| name: Test Torch Models | |
| on: | |
| push: | |
| branches: | |
| - main | |
| paths: | |
| - ".github/workflows/test_torch_models.yml" | |
| - "pytest_iree/**" | |
| - "torch_models/**" | |
| pull_request: | |
| paths: | |
| - ".github/workflows/test_torch_models.yml" | |
| - "pytest_iree/**" | |
| - "torch_models/**" | |
| workflow_dispatch: | |
| schedule: | |
| # Runs at 3:00 PM UTC, which is 8:00 AM PST | |
| - cron: "0 15 * * *" | |
| concurrency: | |
| # A PR number if a pull request and otherwise the commit hash. This cancels | |
| # queued and in-progress runs for the same PR (presubmit) or commit | |
| # (postsubmit). The workflow name is prepended to avoid conflicts between | |
| # different workflows. | |
| group: ${{ github.workflow }}-${{ github.event.number || github.sha }} | |
| cancel-in-progress: true | |
| jobs: | |
| test-torch-models: | |
| if: ${{ github.repository_owner == 'iree-org' || github.event_name != 'schedule' }} | |
| runs-on: | |
| - self-hosted # Must come first. | |
| - persistent-cache | |
| - Linux | |
| - X64 | |
| - threadripper | |
| env: | |
| VENV_DIR: ${{ github.workspace }}/.venv | |
| CACHE_DIR: /home/nod/iree_tests_cache | |
| steps: | |
| - name: Checkout repository | |
| uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2 | |
| with: | |
| lfs: false | |
| # Install Python packages. | |
| - name: Setup Python | |
| uses: actions/setup-python@a309ff8b426b58ec0e2a45f0f869d46889d02405 # v6.2.0 | |
| with: | |
| python-version: "3.11" | |
| - name: Setup Python venv | |
| run: python3 -m venv ${VENV_DIR} | |
| - name: Install IREE nightly release Python packages | |
| run: | | |
| source ${VENV_DIR}/bin/activate | |
| python3 -m pip install -e . | |
| python3 -m pip install -r torch_models/requirements.txt | |
| python3 -m pip install -r torch_models/requirements-iree.txt | |
| # Run tests. | |
| - name: Run Torch models test suite | |
| run: | | |
| source ${VENV_DIR}/bin/activate | |
| pytest \ | |
| torch_models \ | |
| -rpFe \ | |
| --log-cli-level=info \ | |
| -o log_cli=True \ | |
| --durations=0 \ | |
| --timeout=1200 \ | |
| --capture=no \ | |
| --test-file-directory=torch_models \ | |
| --external-file-directory=torch_models \ | |
| --module-directory=torch_models \ | |
| --artifact-directory=${CACHE_DIR}/torch_models/artifacts \ | |
| --job-summary-path=${{ github.workspace }} \ | |
| -m "quick or compstat" | |
| - name: Upload Torch Model Job Summary | |
| if: always() | |
| run: | | |
| cat ${{ github.workspace }}/job_summary.md >> $GITHUB_STEP_SUMMARY |