[ASCEND NPU][CI] Restore execution benchmark (#32)

Restore workflow after confirming benchmark runs successfully.
https://github.com/Ascend/Ascend-CI/actions/runs/16804653153

Author: xuedinge233

.github/workflows/_ascend_npu_benchmark.yml

@@ -90,13 +90,6 @@ jobs:
         run: |
           pip install -r https://raw.githubusercontent.com/Ascend/pytorch/refs/heads/master/requirements.txt
 
-      - name: List torch version
-        id: list-torch-version
-        shell: bash
-        run: |
-          torch_version=$(python -c "import torch; print(torch.__version__)")
-          echo "torch-version=${torch_version}" >> $GITHUB_OUTPUT
-
       - name: Download torch_npu artifact
         uses: actions/download-artifact@v4
         with:
@@ -108,6 +101,13 @@ jobs:
         run: |
           pip install ${{ inputs.torch-npu-artifact }}
 
+      - name: List torch version
+        id: list-torch-version
+        shell: bash
+        run: |
+          torch_version=$(python -c "from importlib.metadata import version;print(version('torch'))")
+          echo "torch-version=${torch_version}" >> $GITHUB_OUTPUT
+
       - name: Install benchmark dependencies
         run: |
           pip install -r benchmark/requirements.txt \

.github/workflows/ascend_npu_test.yml

@@ -133,23 +133,23 @@ jobs:
       torch-artifact: ${{ needs.build-torch.outputs.torch-artifact }}
       torch-npu-artifact: ${{ needs.build.outputs.torch-npu-artifact }}
 
-  # benchmark:
-  #   name: Run benchmarks
-  #   needs:
-  #     - prepare
-  #     - build-torch
-  #     - build
-  #   if: |
-  #     !cancelled() && github.event_name != 'repository_dispatch' &&
-  #     (success() || (needs.build-torch.result == 'skipped' && needs.build.result == 'success'))
-  #   uses: ./.github/workflows/_ascend_npu_benchmark.yml
-  #   with:
-  #     runner: ${{ needs.prepare.outputs.runner }}
-  #     image: ${{ needs.prepare.outputs.image }}
-  #     torch-artifact: ${{ needs.build-torch.outputs.torch-artifact }}
-  #     torch-npu-artifact: ${{ needs.build.outputs.torch-npu-artifact }}
-  #   secrets:
-  #     pr-token: ${{ secrets.ASCEND_RUNNER_TOKEN }}
+  benchmark:
+    name: Run benchmarks
+    needs:
+      - prepare
+      - build-torch
+      - build
+    if: |
+      !cancelled() && github.event_name != 'repository_dispatch' &&
+      (success() || (needs.build-torch.result == 'skipped' && needs.build.result == 'success'))
+    uses: ./.github/workflows/_ascend_npu_benchmark.yml
+    with:
+      runner: ${{ needs.prepare.outputs.runner }}
+      image: ${{ needs.prepare.outputs.image }}
+      torch-artifact: ${{ needs.build-torch.outputs.torch-artifact }}
+      torch-npu-artifact: ${{ needs.build.outputs.torch-npu-artifact }}
+    secrets:
+      pr-token: ${{ secrets.ASCEND_RUNNER_TOKEN }}
 
   torchtune:
     name: Run torchtune