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