Skip to content

Commit 97812f9

Browse files
Added Changes to CI to check if failed_ops.json is empty or not for smoke and opinfo on aten backend (#172)
1 parent 65b7c1a commit 97812f9

File tree

1 file changed

+81
-0
lines changed

1 file changed

+81
-0
lines changed

.github/workflows/failure-test.yml

Lines changed: 81 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,81 @@
1+
name: Failure Test
2+
3+
on:
4+
push:
5+
branches: [main]
6+
pull_request:
7+
branches: [main]
8+
9+
jobs:
10+
smoke-failure-test:
11+
runs-on: 4-core-ubuntu-gpu-t4
12+
steps:
13+
- uses: actions/checkout@v4
14+
- name: Install uv
15+
uses: astral-sh/setup-uv@v3
16+
- name: Set up Python
17+
run: uv python install
18+
- name: Install package and dependencies
19+
run: uv sync --dev
20+
- name: Run smoke test
21+
run: |
22+
uv run python -m BackendBench.scripts.main --suite smoke --backend aten --log-dir smoke_output
23+
- name: Check smoke test failures
24+
run: |
25+
if [ -s "smoke_output/failed_ops.json" ] && [ "$(jq 'length' smoke_output/failed_ops.json)" -ne 0 ]; then
26+
echo "Some operations failed in the smoke test."
27+
cat "smoke_output/failed_ops.json"
28+
exit 1
29+
else
30+
echo "All operations passed the smoke test."
31+
fi
32+
33+
opinfo-failure-test:
34+
runs-on: 4-core-ubuntu-gpu-t4
35+
steps:
36+
- uses: actions/checkout@v4
37+
- name: Install uv
38+
uses: astral-sh/setup-uv@v3
39+
- name: Set up Python
40+
run: uv python install
41+
- name: Install package and dependencies
42+
run: uv sync --dev
43+
- name: Run opinfo test
44+
run: |
45+
uv run python -m BackendBench.scripts.main --suite opinfo --backend aten --log-dir opinfo_output
46+
- name: Check opinfo test failures
47+
run: |
48+
if [ -s "opinfo_output/failed_ops.json" ] && [ "$(jq 'length' opinfo_output/failed_ops.json)" -ne 0 ]; then
49+
echo "Some operations failed in the opinfo test."
50+
cat "opinfo_output/failed_ops.json"
51+
exit 1
52+
else
53+
echo "All operations passed the opinfo test."
54+
fi
55+
56+
facto-failure-test:
57+
runs-on: 4-core-ubuntu-gpu-t4
58+
steps:
59+
- uses: actions/checkout@v4
60+
- name: Install uv
61+
uses: astral-sh/setup-uv@v3
62+
- name: Set up Python
63+
run: uv python install
64+
- name: Install package and dependencies
65+
run: uv sync --dev
66+
- name: Clone FACTO source
67+
run: git clone https://github.com/pytorch-labs/FACTO.git
68+
- name: Build and install FACTO
69+
run: cd FACTO && uv pip install .
70+
- name: Run FACTO test
71+
run: |
72+
uv run python -m BackendBench.scripts.main --suite facto --backend aten --ops "add.Tensor" --log-dir facto_output
73+
- name: Check FACTO test failures
74+
run: |
75+
if [ -s "facto_output/failed_ops.json" ] && [ "$(jq 'length' facto_output/failed_ops.json)" -ne 0 ]; then
76+
echo "Some operations failed in the FACTO test."
77+
cat "facto_output/failed_ops.json"
78+
exit 1
79+
else
80+
echo "All operations passed the FACTO test."
81+
fi

0 commit comments

Comments
 (0)