update

Author: shink

.github/workflows/_ascend_npu_benchmark.yml

@@ -108,6 +108,13 @@ jobs:
         run: |
           pip install ${{ inputs.torch-artifact }}
 
+      - 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:
@@ -121,7 +128,9 @@ jobs:
 
       - name: Install benchmark dependencies
         run: |
-          pip install -r benchmark/requirements.txt numpy==1.*
+          pip install -r benchmark/requirements.txt \
+              torch==${{ steps.list-torch-version.outputs.torch-version }} \
+              numpy==1.*
           python benchmark/install.py --userbenchmark test_bench --continue_on_fail
 
       - name: Install project dependencies

.github/workflows/_ascend_npu_ut.yml

@@ -122,6 +122,14 @@ jobs:
         run: |
           pip install ${{ inputs.torch-npu-artifact }}
 
+      - name: Show environment info
+        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'
+
       # TODO(shink): Skip
       - name: Do the test
         continue-on-error: true