Skip to content

Commit 2c5f24f

Browse files
committed
test
1 parent 23d8acf commit 2c5f24f

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

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

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -41,9 +41,6 @@ jobs:
4141
"""\
4242
import flag_gems\
4343
flag_gems.enable(record=False, once=True, path='"'"'benchmark/oplist.log'"'"')""")' benchmark/torchbenchmark/__init__.py
44-
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 output1.json --models hf_GPT2
46-
python benchmark/run_benchmark.py test_bench --accuracy --device cuda --test eval --output output2.json --models resnet50
47-
python benchmark/run_benchmark.py test_bench --accuracy --device cuda --test eval --output output3.json --models hf_T5
48-
python benchmark/run_benchmark.py test_bench --accuracy --device cuda --test eval --output output4.json --models dcgan
44+
python benchmark/install.py
45+
python benchmark/run_benchmark.py test_bench --accuracy --device cuda --test eval --output output.json
4946
sed -i '/self\.worker\.run($/,/^[[:space:]]*flag_gems\.enable.*oplist\.log.*""")/d' benchmark/torchbenchmark/__init__.py

0 commit comments

Comments
 (0)