Skip to content

Commit fd0ad35

Browse files
committed
Restored TODO
1 parent cd276bf commit fd0ad35

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

scripts/test-triton.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -321,6 +321,7 @@ run_inductor_tests() {
321321

322322
pip install pyyaml pandas scipy numpy psutil pyre_extensions torchrec
323323

324+
# TODO: Find the fastest Hugging Face model
324325
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=$(pwd -P)/inductor_log.csv
325326

326327
cat inductor_log.csv

0 commit comments

Comments
 (0)