update

Author: shink

.github/workflows/_ascend_npu_benchmark.yml

@@ -70,12 +70,21 @@ jobs:
           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 }}
           torch_npu_version=$(python -c "import torch_npu; print(torch_npu.__version__)")
+          npu_is_available=$(python -c "import torch; print(torch.npu.is_available())")
+          npu_count=$(python -c "import torch; print(torch.npu.device_count())")
           echo "Installed torch_npu: ${torch_npu_version}"
+          echo "NPU is available: ${npu_is_available}"
+          echo "NPU count: ${npu_count}"
 
       - name: Checkout benchmark
         uses: actions/checkout@v4