Skip to content

Commit 88150df

Browse files
committed
test
1 parent f864601 commit 88150df

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

.github/workflows/_ascend_npu_benchmark.yml

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,10 @@ jobs:
8383
- name: Checkout benchmark
8484
uses: actions/checkout@v4
8585
with:
86-
repository: pytorch/benchmark
86+
# TODO(shink): https://github.com/pytorch/benchmark/pull/2592
87+
# repository: pytorch/benchmark
88+
repository: shink/benchmark
89+
ref: feat/test_bench/config
8790
path: benchmark
8891

8992
# TODO
@@ -152,18 +155,16 @@ jobs:
152155
pip list | grep -E 'torch|numpy'
153156
154157
- name: Run benchmarks
155-
working-directory: benchmark
156158
run: |
157-
python run_benchmark.py test_bench --accuracy --device npu --test eval \
158-
--output ascend_npu_benchmark.json
159+
python benchmark/run_benchmark.py test_bench --config ascend_npu/torchbenchmark-config.yml
159160
160161
- name: Upload the benchmark report file
161162
id: upload-report
162163
uses: actions/upload-artifact@v4
163164
with:
164165
name: ascend_npu_benchmark.json
165166
path: benchmark/ascend_npu_benchmark.json
166-
if-no-files-found: error
167+
if-no-files-found: warn
167168
retention-days: 3
168169
overwrite: true
169170

0 commit comments

Comments
 (0)