Skip to content

Commit ba74488

Browse files
committed
fix some errors
1 parent c215c3b commit ba74488

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

.github/workflows/ci-temp-model-test.yaml

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,5 +42,10 @@ jobs:
4242
import flag_gems\
4343
flag_gems.enable(record=False, once=True, path='"'"'benchmark/oplist.log'"'"')""")' benchmark/torchbenchmark/__init__.py
4444
python benchmark/install.py models hf_GPT2 resnet50 hf_T5 dcgan
45-
python benchmark/run_benchmark.py test_bench --accuracy --device cuda --test eval --output output.json --models hf_GPT2 resnet50 hf_T5 dcgan
45+
python benchmark/run_benchmark.py test_bench \
46+
--device cuda \
47+
--test eval \
48+
--output output.json \
49+
--models hf_GPT2 resnet50 hf_T5 dcgan \
50+
--accuracy
4651
sed -i '/self\.worker\.run($/,/^[[:space:]]*flag_gems\.enable.*oplist\.log.*""")/d' benchmark/torchbenchmark/__init__.py

0 commit comments

Comments
 (0)