File tree Expand file tree Collapse file tree 1 file changed +6
-5
lines changed
Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Original file line number Diff line number Diff line change 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
You can’t perform that action at this time.
0 commit comments