Skip to content

Commit 43e70d9

Browse files
committed
Add a standalone e2e test, r2
1 parent df03c33 commit 43e70d9

File tree

2 files changed

+18
-1
lines changed

2 files changed

+18
-1
lines changed

.github/workflows/build-test-reusable.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,7 @@ jobs:
145145
- name: Run E2E test
146146
run: |
147147
#FIXME move back test when ready
148-
scripts/inductor_xpu_test.sh huggingface float32 inference accuracy xpu 0 static 1 0 AlbertForMaskedLM
148+
scripts/test-inductor.sh
149149
exit 1
150150
151151
- name: Create test-triton command line
@@ -190,6 +190,7 @@ jobs:
190190
${{ env.TRITON_TEST_CMD }} --instrumentation --skip-pip-install
191191
192192
- name: Get transformers version
193+
if: inputs.env_manager == 'base'
193194
run: |
194195
cd pytorch
195196
TRANSFORMERS_VERSION="$(<.ci/docker/ci_commit_pins/huggingface.txt)"

scripts/test-inductor.sh

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
#/bin/sh
2+
3+
test -d pytorch || {
4+
git clone https://github.com/pytorch/pytorch
5+
cd pytorch
6+
TRANSFORMERS_VERSION=$(.ci/docker/ci_commit_pins/huggingface.txt)
7+
pip install transformers=$TRANSFORMERS_VERSION
8+
python -c "import transformers; print(transformers.__version__)"
9+
}
10+
11+
pip install pyyaml pandas scipy numpy psutil pyre_extensions torchrec
12+
13+
ZE_AFFINITY_MASK=0 python pytorch/benchmarks/dynamo/huggingface.py --accuracy --float32 -dxpu -n10 --no-skip --dashboard --inference --freezing --total-partitions 1 --partition-id 0 --only AlbertForMaskedLM --backend=inductor --timeout=4800 --output=inductor_log.csv
14+
15+
cat inductor_log.csv
16+
grep AlbertForMaskedLM inductor_log.csv | grep -q ,pass,

0 commit comments

Comments
 (0)