Skip to content

Commit 33a63cc

Browse files
committed
update
1 parent 87dd327 commit 33a63cc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/workflows/_ascend_npu_benchmark.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ jobs:
9090
- name: Install benchmark dependencies
9191
run: |
9292
pip install -r benchmark/requirements.txt --constraint ascend_npu/requirements.txt "numpy==1.*"
93-
python benchmark/install.py --userbenchmark test_bench --models BERT_pytorch hf_GPT2
93+
python benchmark/install.py --userbenchmark test_bench
9494
9595
- name: Install project dependencies
9696
run: |
@@ -108,7 +108,6 @@ jobs:
108108
working-directory: benchmark
109109
run: |
110110
python run_benchmark.py test_bench --accuracy --device npu --test eval \
111-
--models BERT_pytorch,hf_GPT2 \
112111
--output ascend_npu_benchmark.json
113112
114113
- name: Upload the output file
@@ -132,6 +131,7 @@ jobs:
132131
output_path=$(realpath benchmark/ascend_npu_benchmark.json)
133132
python .ci/benchmark.py ${output_path}
134133
134+
# TODO(shink)
135135
- name: Update README.md
136136
if: ${{ github.event_name == 'push' }}
137137
run: |

0 commit comments

Comments
 (0)