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