-
Notifications
You must be signed in to change notification settings - Fork 44
43 lines (40 loc) · 1.2 KB
/
inductor-tests.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
name: Pytorch inductor tests
on:
workflow_dispatch:
inputs:
pytorch_ref:
description: PyTorch ref, keep empty for default
type: string
default: ""
suite:
description: Space separated lists of test suites, all if empty
type: string
default: >-
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
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.9"
permissions: read-all
jobs:
run_tests:
name: Run inductor tests
uses: ./.github/workflows/inductor-tests-reusable.yml
with:
pytorch_repo: ${{ inputs.pytorch_repo }}
python_version: ${{ inputs.python_version }}
runner_label: ${{ inputs.runner_label }}
suite: ${{ inputs.suite }}
pytorch_ref: ${{ inputs.pytorch_ref }}