Skip to content

Commit 9880bc1

Browse files
committed
test
1 parent ba74488 commit 9880bc1

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

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

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -42,10 +42,5 @@ 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 \
46-
--device cuda \
47-
--test eval \
48-
--output output.json \
49-
--models hf_GPT2 resnet50 hf_T5 dcgan \
50-
--accuracy
45+
python run_benchmark.py test_bench --accuracy --device cuda --test eval --output output.json
5146
sed -i '/self\.worker\.run($/,/^[[:space:]]*flag_gems\.enable.*oplist\.log.*""")/d' benchmark/torchbenchmark/__init__.py

0 commit comments

Comments
 (0)