Skip to content

Commit 84b3480

Browse files
committed
add generated test script
1 parent 8fcbac1 commit 84b3480

File tree

2 files changed

+26
-1
lines changed

2 files changed

+26
-1
lines changed

.github/workflows/ci_build_test_cpp_linux.yml

+5-1
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ jobs:
3131
- name: "Setting up Python"
3232
uses: actions/setup-python@75f3110429a8c05be0e1bf360334e4cced2b63fa # v2.3.3
3333
with:
34-
python-version: "3.11"
34+
python-version: "3.10"
3535

3636
- name: Sync source deps
3737
run: |
@@ -59,3 +59,7 @@ jobs:
5959
# Some things put stuff in cache with weird, root read-only
6060
# permissions. Take them back.
6161
sudo chown -R "$(whoami)" "${cache_dir}"
62+
63+
- name: Run generated tests
64+
run: |
65+
bash build_tools/ci/run_paritybench_test.sh
+21
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
1+
#!/bin/bash
2+
3+
git clone https://github.com/jansel/pytorch-jit-paritybench.git
4+
5+
cd pytorch-jit-paritybench
6+
7+
git checkout 7e55a422588c1d1e00f35a3d3a3ff896cce59e18
8+
pip install -r requirements.txt
9+
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
10+
pip install expecttest
11+
12+
cd ..
13+
14+
pip install ./compiler
15+
pip install ./runtime
16+
pip install . --no-deps
17+
source build/iree/.env
18+
19+
python python/test/generated/main.py --tests-dir ./pytorch-jit-paritybench --limit 100 -j 8
20+
21+

0 commit comments

Comments
 (0)