use benchmark configuration

Author: shink

.github/workflows/_ascend_npu_benchmark.yml

@@ -127,6 +127,11 @@ jobs:
         run: |
           pip install ${{ inputs.torch-npu-artifact }}
 
+      - name: Install torchvision and torchaudio
+        run: |
+          # pip install --pre torchvision torchaudio --no-deps --index-url https://download.pytorch.org/whl/nightly/cpu
+          pip install --pre torchvision torchaudio --no-deps --index-url https://download.pytorch.org/whl/test/cpu
+
       - name: Install benchmark dependencies
         run: |
           pip install -r benchmark/requirements.txt \
@@ -137,20 +142,13 @@ jobs:
         run: |
           python benchmark/install.py --userbenchmark test_bench --continue_on_fail
 
-      - name: Install nightly torchvision and torchaudio
-        run: |
-          pip install --pre torchvision torchaudio --no-deps --index-url https://download.pytorch.org/whl/nightly/cpu
-
       - name: Install project dependencies
         run: |
           pip install -r requirements.txt
 
-      - name: Show environment info
+      - name: List installed python packages
+        if: ${{ always() }}
         run: |
-          npu_is_available=$(python -c "import torch; print(torch.npu.is_available())")
-          npu_count=$(python -c "import torch; print(torch.npu.device_count())")
-          echo "NPU is available: ${npu_is_available}"
-          echo "NPU count: ${npu_count}"
           pip list | grep -E 'torch|numpy'
 
       - name: Run benchmarks
@@ -166,7 +164,7 @@ jobs:
           name: ascend_npu_benchmark.json
           path: benchmark/ascend_npu_benchmark.json
           if-no-files-found: error
-          retention-days: 1
+          retention-days: 3
           overwrite: true
 
       - name: Write to workflow job summary

.github/workflows/_ascend_npu_build_torch_npu.yml

@@ -73,11 +73,7 @@ jobs:
       - name: Checkout torch_npu
         uses: actions/checkout@v4
         with:
-          # TODO(shink): Use Ascend/pytorch once this pr merged:
-          # https://gitee.com/ascend/pytorch/pulls/12854
-          # repository: Ascend/pytorch
-          repository: shink/torchnpu
-          ref: feat/autoload
+          repository: Ascend/pytorch
           submodules: recursive
           path: torch_npu
 

.github/workflows/_ascend_npu_ut.yml

@@ -75,11 +75,7 @@ jobs:
       - name: Checkout torch_npu
         uses: actions/checkout@v4
         with:
-          # TODO(shink): Use Ascend/pytorch once this pr merged:
-          # https://gitee.com/ascend/pytorch/pulls/12854
-          # repository: Ascend/pytorch
-          repository: shink/torchnpu
-          ref: feat/autoload
+          repository: Ascend/pytorch
           path: torch_npu
 
       - name: Install pip dependencies

ascend_npu/matadata.yml

@@ -0,0 +1,10 @@
+device: "npu"
+backend_extension: "torch_npu"
+link: https://github.com/Ascend/pytorch
+models:
+  - model: BERT_pytorch
+    tests: [ "eval" ]
+    batch_size: 1
+
+  - model: yolov3
+    skip: true