Skip to content

Commit e03979d

Browse files
committed
update
1 parent 2efbf01 commit e03979d

File tree

1 file changed

+12
-11
lines changed

1 file changed

+12
-11
lines changed

.github/workflows/_ascend_npu_benchmark.yml

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -61,17 +61,6 @@ jobs:
6161
apt update
6262
apt install --no-install-recommends -y git gcc g++ make cmake ninja-build curl
6363
64-
- name: Install torch_npu
65-
run: |
66-
pip install torch-npu --index-url https://pypi.org/simple/
67-
68-
- name: Install torch_npu dependencies
69-
run: |
70-
echo $(pwd)
71-
echo $(ls -la)
72-
curl -fsSL -o ascend_npu/requirements.txt https://raw.githubusercontent.com/Ascend/pytorch/refs/heads/master/requirements.txt
73-
pip install -r ascend_npu/requirements.txt
74-
7564
- name: Checkout
7665
uses: actions/checkout@v4
7766

@@ -81,9 +70,21 @@ jobs:
8170
repository: pytorch/benchmark
8271
path: benchmark
8372

73+
# TODO
74+
- name: Install torch_npu
75+
run: |
76+
pip install torch-npu --index-url https://pypi.org/simple
77+
78+
- name: Install torch_npu dependencies
79+
working-directory: ascend_npu
80+
run: |
81+
curl -fsSL -O https://raw.githubusercontent.com/Ascend/pytorch/refs/heads/master/requirements.txt
82+
pip install -r requirements.txt
83+
8484
- name: Install benchmark dependencies
8585
run: |
8686
ls -al ascend_npu
87+
echo "\n"
8788
pip install -r benchmark/requirements.txt --constraint ascend_npu/requirements.txt
8889
python benchmark/install.py --userbenchmark test_bench --models BERT_pytorch hf_GPT2
8990

0 commit comments

Comments
 (0)