Skip to content

Commit 0854741

Browse files
committed
update
1 parent 0c96eba commit 0854741

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

.github/workflows/_ascend_npu_benchmark.yml

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,12 @@ jobs:
7272
run: |
7373
pip install ${{ inputs.artifact_name }}
7474
75+
- name: Install torch_npu dependencies
76+
working-directory: ascend_npu
77+
run: |
78+
curl -fsSL -O https://raw.githubusercontent.com/Ascend/pytorch/refs/heads/master/requirements.txt
79+
pip install -r requirements.txt
80+
7581
- name: Checkout
7682
uses: actions/checkout@v4
7783

@@ -82,10 +88,9 @@ jobs:
8288
path: benchmark
8389

8490
- name: Install benchmark dependencies
85-
working-directory: benchmark
8691
run: |
87-
pip install -r requirements.txt
88-
python install.py --userbenchmark test_bench --models BERT_pytorch hf_GPT2
92+
pip install -r benchmark/requirements.txt --constraint ascend_npu/requirements.txt
93+
python benchmark/install.py --userbenchmark test_bench --models BERT_pytorch hf_GPT2 --constraint ascend_npu/requirements.txt
8994
9095
- name: Run benchmarks
9196
working-directory: benchmark

0 commit comments

Comments
 (0)