update

Author: shink

.github/workflows/_ascend_npu_benchmark.yml

@@ -99,14 +99,21 @@ jobs:
       #     artifact: ${{ inputs.torch-artifact }}
 
       - name: Download torch artifact
+        if: ${{ inputs.torch-artifact }}
         uses: actions/download-artifact@v4
         with:
           name: ${{ inputs.torch-artifact }}
 
       - name: Install torch
+        if: ${{ inputs.torch-artifact }}
         run: |
           pip install ${{ inputs.torch-artifact }}
 
+      - name: Install torch_npu dependencies
+        if: ${{ !inputs.torch-artifact }}
+        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

.github/workflows/_ascend_npu_build_torch_npu.yml

@@ -43,19 +43,6 @@ jobs:
     outputs:
       dist-name: ${{ steps.list-dist.outputs.dist-name }}
     steps:
-      - name: Hit
-        if: ${{ inputs.torch-artifact }}
-        run: |
-          set -x
-          echo ${{ inputs.torch-artifact }}
-          exit 1
-
-      - name: Hit
-        run: |
-          set -x
-          echo ${{ inputs.torch-artifact }}
-          exit 1
-
       - name: Config mirrors
         run: |
           sed -i 's|ports.ubuntu.com|mirrors.tuna.tsinghua.edu.cn|g' /etc/apt/sources.list
@@ -99,11 +86,13 @@ jobs:
       #     artifact: ${{ inputs.torch-artifact }}
 
       - name: Download torch artifact
+        if: ${{ inputs.torch-artifact }}
         uses: actions/download-artifact@v4
         with:
           name: ${{ inputs.torch-artifact }}
 
       - name: Install torch
+        if: ${{ inputs.torch-artifact }}
         run: |
           pip install ${{ inputs.torch-artifact }}
 

.github/workflows/_ascend_npu_ut.yml

@@ -102,11 +102,13 @@ jobs:
       #     artifact: ${{ inputs.torch-artifact }}
 
       - name: Download torch artifact
+        if: ${{ inputs.torch-artifact }}
         uses: actions/download-artifact@v4
         with:
           name: ${{ inputs.torch-artifact }}
 
       - name: Install torch
+        if: ${{ inputs.torch-artifact }}
         run: |
           pip install ${{ inputs.torch-artifact }}