File tree Expand file tree Collapse file tree 2 files changed +18
-1
lines changed
Expand file tree Collapse file tree 2 files changed +18
-1
lines changed Original file line number Diff line number Diff 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)"
Original file line number Diff line number Diff line change 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,
You can’t perform that action at this time.
0 commit comments