PyTorch Flex Decoding, PVC (Max 1100), pinned #1864
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: PyTorch inductor tests on Linux | |
| run-name: ${{ inputs.run_name }} | |
| on: | |
| workflow_dispatch: | |
| inputs: | |
| pytorch_ref: | |
| description: PyTorch ref, keep empty for default | |
| type: string | |
| default: "" | |
| suite: | |
| description: Space separated lists of test suites or "all" | |
| type: string | |
| default: "" | |
| runner_label: | |
| description: Runner label, keep empty for default | |
| type: string | |
| default: "" | |
| pytorch_repo: | |
| description: PyTorch repo | |
| type: string | |
| default: "pytorch/pytorch" | |
| python_version: | |
| description: Python version | |
| type: string | |
| default: "3.10" | |
| run_name: | |
| description: Custom run name | |
| type: string | |
| default: "PyTorch inductor tests on Linux" | |
| pull_request: | |
| branches: | |
| - main | |
| paths: | |
| - .github/pins/pytorch.txt | |
| - .github/workflows/inductor-tests.yml | |
| - .github/workflows/inductor-tests-reusable.yml | |
| permissions: read-all | |
| env: | |
| DEFAULT_SUITE: >- | |
| inductor/test_kernel_benchmark.py | |
| inductor/test_codegen_triton.py | |
| inductor/test_triton_extension_backend.py | |
| inductor/test_triton_heuristics.py | |
| inductor/test_triton_wrapper.py | |
| inductor/test_triton_kernels.py | |
| inductor/test_gpu_cpp_wrapper.py | |
| inductor/test_aot_inductor.py | |
| inductor/test_select_algorithm.py | |
| inductor/test_max_autotune.py | |
| inductor/test_compile_subprocess.py | |
| jobs: | |
| compute-params: | |
| runs-on: linux | |
| outputs: | |
| suite: ${{ steps.set_suite.outputs.suite }} | |
| steps: | |
| - id: set_suite | |
| run: | | |
| SUITE="${{ inputs.suite }}" | |
| if [ -z "${SUITE}" ]; then | |
| echo "Using default suite" | |
| SUITE="${DEFAULT_SUITE}" | |
| fi | |
| echo "suite=${SUITE}" >> "$GITHUB_OUTPUT" | |
| run_tests: | |
| name: Run inductor tests | |
| needs: compute-params | |
| uses: ./.github/workflows/inductor-tests-reusable.yml | |
| with: | |
| pytorch_repo: ${{ inputs.pytorch_repo || 'pytorch/pytorch' }} | |
| python_version: ${{ inputs.python_version || '3.10' }} | |
| runner_label: ${{ inputs.runner_label || '' }} | |
| suite: ${{ needs.compute-params.outputs.suite }} | |
| pytorch_ref: ${{ inputs.pytorch_ref || '' }} |