Skip to content

Commit adcf6a7

Browse files
authored
Fix benchmark errors (#14)
1 parent a1695ac commit adcf6a7

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

.github/workflows/_ascend_npu_benchmark.yml

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,10 @@ on:
1919
required: true
2020
type: string
2121
description: 'The torch_npu distribution artifact name'
22+
secrets:
23+
pr-token:
24+
description: 'A token used to create a pull request'
25+
required: true
2226

2327
# Bash shells do not use ~/.profile or ~/.bashrc so these shells need to be explicitly
2428
# declared as "shell: bash -el {0}" on steps that need to be properly activated.
@@ -31,6 +35,8 @@ jobs:
3135
test:
3236
name: run benchmarks for torch_npu
3337
runs-on: ${{ inputs.runner }}
38+
env:
39+
HF_ENDPOINT: https://hf-mirror.com
3440
container:
3541
image: ${{ inputs.image }}
3642
volumes:
@@ -100,8 +106,6 @@ jobs:
100106
run: |
101107
pip install -r benchmark/requirements.txt --constraint ascend_npu/requirements.txt "numpy==1.*"
102108
python benchmark/install.py --userbenchmark test_bench --continue_on_fail
103-
env:
104-
HF_ENDPOINT: https://hf-mirror.com
105109
106110
- name: Install project dependencies
107111
run: |
@@ -150,6 +154,7 @@ jobs:
150154
if: ${{ steps.update-readme.outputs.changed == 'true' }}
151155
uses: peter-evans/create-pull-request@v7
152156
with:
157+
token: ${{ secrets.pr-token }}
153158
add-paths: README.md
154159
branch: ascend-npu/benchmark
155160
title: "[Ascend NPU] Update torchbenchmark report in README.md"

.github/workflows/ascend_npu_test.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,3 +122,5 @@ jobs:
122122
image: ${{ needs.prepare.outputs.image }}
123123
device: ${{ needs.prepare.outputs.device }}
124124
artifact_name: ${{ needs.build.outputs.artifact_name }}
125+
secrets:
126+
pr-token: ${{ secrets.COSDT_BOT_TOKEN }}

0 commit comments

Comments
 (0)