Skip to content

Commit 4462554

Browse files
committed
Workflow to test Triton with pip dependencies
1 parent b5a791e commit 4462554

File tree

1 file changed

+56
-0
lines changed

1 file changed

+56
-0
lines changed

.github/workflows/pip-test.yml

Lines changed: 56 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,56 @@
1+
name: Test with pip
2+
3+
on:
4+
workflow_dispatch:
5+
6+
# FIXME: update before merging
7+
pull_request:
8+
branches:
9+
- main
10+
push:
11+
branches:
12+
- main
13+
14+
permissions: read-all
15+
16+
env:
17+
PYTHON_VERSION: '3.9'
18+
TRITON_TEST_CMD: "scripts/test-triton.sh --skip-pytorch-install"
19+
20+
jobs:
21+
tests:
22+
name: Tests
23+
runs-on:
24+
- rolling
25+
- runner-0.0.20
26+
steps:
27+
- name: Install Python
28+
uses: actions/setup-python@v5
29+
with:
30+
python-version: ${{ env.PYTHON_VERSION }}
31+
32+
- name: Install wheels
33+
uses: ./.github/actions/install-wheels
34+
with:
35+
gh_token: ${{ secrets.GITHUB_TOKEN }}
36+
python_version: ${{ env.PYTHON_VERSION }}
37+
38+
- name: Install Triton runtime dependencies
39+
run: |
40+
pip install intel-sycl-rt intel-pti
41+
42+
- name: Run core tests
43+
run: |
44+
${{ env.TRITON_TEST_CMD }} --core
45+
46+
- name: Run interpreter tests
47+
run: |
48+
${{ env.TRITON_TEST_CMD }} --interpreter --skip-pip-install
49+
50+
- name: Run Tutorials
51+
run: |
52+
${{ env.TRITON_TEST_CMD }} --tutorial --skip-pip-install
53+
54+
- name: Run inductor test
55+
run: |
56+
${{ env.TRITON_TEST_CMD }} --inductor --skip-pip-install

0 commit comments

Comments
 (0)