tmp

Author: shink

.github/workflows/_ascend_npu_build_torch_npu.yml

@@ -12,7 +12,7 @@ on:
         type: string
         description: 'The docker image which will be used to build'
       torch-artifact:
-        required: true
+        required: false
         type: string
         description: 'The distribution artifact name of torch'
     outputs:
@@ -43,6 +43,10 @@ jobs:
     outputs:
       dist-name: ${{ steps.list-dist.outputs.dist-name }}
     steps:
+      - run: |
+          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

.github/workflows/ascend_npu_test.yml

@@ -96,14 +96,15 @@ jobs:
       # TODO(shink): List ghstack PR's ref
       - name: List ref to the PyTorch branch
         id: list-ref
+        if: ${{ github.event_name == 'repository_dispatch' }}
         run: |
-          # echo "ref=refs/pull/${{ github.event.client_payload.pull_request.number }}/merge" >> $GITHUB_OUTPUT
-          echo "ref=refs/heads/main" >> $GITHUB_OUTPUT
+          echo "ref=refs/pull/${{ github.event.client_payload.pull_request.number }}/merge" >> $GITHUB_OUTPUT
 
   build-torch:
     name: Build torch
     needs:
       - prepare
+    if: ${{ needs.prepare.outputs.ref }}
     uses: ./.github/workflows/_ascend_npu_build_torch.yml
     with:
       runner: ${{ needs.prepare.outputs.runner }}
@@ -119,7 +120,7 @@ jobs:
     with:
       runner: ${{ needs.prepare.outputs.runner }}
       image: ${{ needs.prepare.outputs.image }}
-      torch-artifact: ${{ needs.build-torch.outputs.torch-artifact }}
+      torch-artifact: ${{ needs.build-torch.outputs.torch-artifact || null }}
 
   test:
     name: Test torch_npu