update

Author: shink

.github/actions/install-torch/action.yml

@@ -3,6 +3,7 @@ description: "Install torch built from the build workflow"
 inputs:
   torch-artifact:
     description: "The distribution artifact name of torch"
+    type: string
     required: true
 outputs:
   torch-version:

.github/workflows/_ascend_npu_benchmark.yml

@@ -91,11 +91,29 @@ jobs:
           repository: pytorch/benchmark
           path: benchmark
 
+      # TODO
+      #- name: Install torch
+      #  id: install-torch
+      #  uses: ./.github/actions/install-torch
+      #  with:
+      #    torch-artifact: ${{ inputs.torch-artifact }}
+
+      - name: Download torch artifact
+        uses: actions/download-artifact@v4
+        with:
+          name: ${{ inputs.torch-artifact }}
+
       - name: Install torch
+        shell: bash
+        run: |
+          pip install ${{ inputs.torch-artifact }}
+
+      - name: List torch version
         id: install-torch
-        uses: ./.github/actions/install-torch
-        with:
-          torch-artifact: ${{ inputs.torch-artifact }}
+        shell: bash
+        run: |
+          torch_version=$(python -c "import torch; print(torch.__version__)")
+          echo "torch_version=${torch_version}" >> $GITHUB_OUTPUT
 
       # We must use numpy 1.x
       - name: Generate pip's constraint file

.github/workflows/_ascend_npu_build_torch.yml

@@ -59,33 +59,6 @@ jobs:
         run: |
           git config --global --add safe.directory "$GITHUB_WORKSPACE"
 
-      - name: Show
-        run: |
-          set -x
-          pwd
-          ls -al
-          echo ${{ github.workspace }}
-          echo $GITHUB_WORKSPACE
-
-      - name: Checkout
-        uses: actions/checkout@v4
-
-      - name: Show
-        run: |
-          set -x
-          pwd
-          ls -al
-
-      - name: Install torch
-        id: install-torch
-        uses: ./.github/actions/install-torch
-        with:
-          torch-artifact: xxx
-
-      - name: Exit
-        run: |
-          exit 1
-
       - name: Checkout PyTorch
         uses: actions/checkout@v4
         with:

.github/workflows/_ascend_npu_build_torch_npu.yml

@@ -73,11 +73,29 @@ jobs:
           submodules: recursive
           path: torch_npu
 
+      # TODO
+      #- name: Install torch
+      #  id: install-torch
+      #  uses: ./.github/actions/install-torch
+      #  with:
+      #    torch-artifact: ${{ inputs.torch-artifact }}
+
+      - name: Download torch artifact
+        uses: actions/download-artifact@v4
+        with:
+          name: ${{ inputs.torch-artifact }}
+
       - name: Install torch
+        shell: bash
+        run: |
+          pip install ${{ inputs.torch-artifact }}
+
+      - name: List torch version
         id: install-torch
-        uses: ./.github/actions/install-torch
-        with:
-          torch-artifact: ${{ inputs.torch-artifact }}
+        shell: bash
+        run: |
+          torch_version=$(python -c "import torch; print(torch.__version__)")
+          echo "torch_version=${torch_version}" >> $GITHUB_OUTPUT
 
       - name: Generate pip's constraint file
         working-directory: torch_npu

.github/workflows/_ascend_npu_ut.yml

@@ -87,11 +87,29 @@ jobs:
           ref: feat/autoload
           path: torch_npu
 
+      # TODO
+      #- name: Install torch
+      #  id: install-torch
+      #  uses: ./.github/actions/install-torch
+      #  with:
+      #    torch-artifact: ${{ inputs.torch-artifact }}
+
+      - name: Download torch artifact
+        uses: actions/download-artifact@v4
+        with:
+          name: ${{ inputs.torch-artifact }}
+
       - name: Install torch
+        shell: bash
+        run: |
+          pip install ${{ inputs.torch-artifact }}
+
+      - name: List torch version
         id: install-torch
-        uses: ./.github/actions/install-torch
-        with:
-          torch-artifact: ${{ inputs.torch-artifact }}
+        shell: bash
+        run: |
+          torch_version=$(python -c "import torch; print(torch.__version__)")
+          echo "torch_version=${torch_version}" >> $GITHUB_OUTPUT
 
       - name: Generate pip's constraint file
         working-directory: torch_npu