update

Author: shink

.github/workflows/_ascend_npu_benchmark.yml

@@ -61,28 +61,19 @@ jobs:
           apt update
           apt install --no-install-recommends -y git gcc g++ make cmake ninja-build
 
-      - name: Checkout
-        uses: actions/checkout@v4
-
-      - name: Download distribution artifact
+      - name: Download ${{ inputs.artifact_name }}
         uses: actions/download-artifact@v4
         with:
           name: ${{ inputs.artifact_name }}
           path: ascend_npu
 
-      - name: Install torch
-        working-directory: ascend_npu
-        run: |
-          pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
-
       - name: Install torch_npu
         working-directory: ascend_npu
         run: |
           pip install ${{ inputs.artifact_name }}
-          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}"
+
+      - name: Checkout
+        uses: actions/checkout@v4
 
       - name: Checkout benchmark
         uses: actions/checkout@v4