refactor(cache): faster lookup using list_repo_tree #2100
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
| name: Optimum neuron / Test INF2 Export | |
| on: | |
| push: | |
| branches: [ main ] | |
| paths: | |
| - "pyproject.toml" | |
| - "optimum/commands/export/neuronx.py" | |
| - "optimum/exporters/**.py" | |
| - "optimum/neuron/modeling_base.py" | |
| - "optimum/neuron/modeling_diffusion.py" | |
| - "optimum/neuron/modeling_traced.py" | |
| - "optimum/neuron/modeling_seq2seq.py" | |
| - "optimum/neuron/models/inference/bert/**.py" | |
| - "optimum/neuron/models/inference/clip/**.py" | |
| - "optimum/neuron/models/inference/flux/**.py" | |
| - "optimum/neuron/models/inference/t5/**.py" | |
| - "optimum/neuron/models/inference/whisper/**.py" | |
| - "optimum/neuron/models/inference/yolos/**.py" | |
| - "optimum/neuron/utils/**.py" | |
| - "tests/exporters/**.py" | |
| - ".github/workflows/test_inf2_export.yml" | |
| pull_request: | |
| branches: [ main ] | |
| paths: | |
| - "pyproject.toml" | |
| - "optimum/commands/export/neuronx.py" | |
| - "optimum/exporters/**.py" | |
| - "optimum/neuron/modeling_base.py" | |
| - "optimum/neuron/modeling_diffusion.py" | |
| - "optimum/neuron/modeling_traced.py" | |
| - "optimum/neuron/modeling_seq2seq.py" | |
| - "optimum/neuron/models/inference/bert/**.py" | |
| - "optimum/neuron/models/inference/clip/**.py" | |
| - "optimum/neuron/models/inference/flux/**.py" | |
| - "optimum/neuron/models/inference/t5/**.py" | |
| - "optimum/neuron/models/inference/whisper/**.py" | |
| - "optimum/neuron/models/inference/yolos/**.py" | |
| - "optimum/neuron/utils/**.py" | |
| - "tests/exporters/**.py" | |
| - ".github/workflows/test_inf2_export.yml" | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }} | |
| cancel-in-progress: true | |
| jobs: | |
| do-the-job: | |
| name: Run INF2 export tests | |
| runs-on: | |
| group: aws-inf2-8xlarge | |
| steps: | |
| - name: Checkout | |
| uses: actions/checkout@v4 | |
| - name: Install Neuronx runtime | |
| uses: ./.github/actions/install_neuronx_runtime | |
| - name: Prepare virtual environment | |
| uses: ./.github/actions/prepare_venv | |
| - name: Install optimum-neuron | |
| uses: ./.github/actions/install_optimum_neuron | |
| - name: Install diffusers dependencies | |
| run: | | |
| source aws_neuron_venv_pytorch/bin/activate | |
| pip install .[diffusers] | |
| - name: Run exporters tests | |
| run: | | |
| source aws_neuron_venv_pytorch/bin/activate | |
| export RANDOM_TASK_PER_MODEL=TRUE | |
| HF_TOKEN=${{ secrets.HF_TOKEN_OPTIMUM_NEURON_CI }} pytest tests/exporters --ignore=tests/exporters/test_cpu_compilation.py --durations=0 --timeout=300 |