update

Author: shink

.github/actions/dependencies-action/action.yml

@@ -29,7 +29,6 @@ inputs:
 runs:
   using: composite
   steps:
-    # TODO(shink): Remove this step after building devel images
     - name: Install system dependencies
       shell: bash
       env:
@@ -40,19 +39,13 @@ runs:
         apt update
         apt install --no-install-recommends -y \
             git \
-            gcc-10 \
-            g++-10 \
+            gcc \
+            g++ \
             make \
             cmake \
             ninja-build
         echo "::endgroup::"
 
-    - name: Create symlinks for gcc and g++
-      shell: bash
-      run: |
-        ln -s /usr/bin/gcc-10 /usr/bin/gcc
-        ln -s /usr/bin/g++-10 /usr/bin/g++
-
     - name: Show versions
       shell: bash
       run: |

.github/workflows/_ascend_npu_build.yml

@@ -29,40 +29,46 @@ jobs:
     runs-on: ${{ inputs.runner }}
     container:
       image: ${{ inputs.image }}
-      volumes:
-        - /home/runner/actions-runner/codes:/root/codes
     outputs:
       dist_name: ${{ steps.list-dist.outputs.dist_name }}
     steps:
-      - name: Prepare the codes
-        run: |
-          cp -rf /root/codes /root/build
+      - name: Install system dependencies
+        uses: ./.github/actions/dependencies-action
 
       - name: Checkout
         uses: actions/checkout@v4
 
-      - name: Install dependencies
-        uses: ./.github/actions/dependencies-action
+      - name: Checkout torch_npu
+        uses: actions/checkout@v4
         with:
-          pip_packages: |
-            - wheel
-          pip_requirements: |
-            - /root/build/npu/pytorch/requirements.txt
+          # TODO(shink): Use ascend/pytorch once this pr merged:
+          # https://gitee.com/ascend/pytorch/pulls/12854
+          repo: shink/torchnpu
+          ref: feat/autoload
+          submodules: recursive
+          path: torch_npu
+
+      - name: Install pip dependencies
+        working-directory: torch_npu
+        run: |
+          pip install wheel
+          pip install -r requirements.txt
 
       - name: List Python version
         id: list-py-version
+        working-directory: torch_npu
         run: |
           py_version=$(python --version | awk '{print $2}' | cut -d '.' -f 1,2)
           echo "py_version=${py_version}" >> $GITHUB_OUTPUT
 
       - name: Build torch_npu
-        working-directory: /root/build/npu/pytorch
+        working-directory: torch_npu
         run: |
           bash ci/build.sh --python=${{ steps.list-py-version.outputs.py_version }}
 
       - name: List distribution package
         id: list-dist
-        working-directory: /root/build/npu/pytorch/dist
+        working-directory: torch_npu/dist
         run: |
           dist_name=$(ls torch_npu*.whl)
           dist_path=$(pwd)/${dist_name}

.github/workflows/ascend_npu_test.yml

@@ -1,6 +1,3 @@
-# Note:
-# Same runner only needs one job named like .*fetch-and-rebase
-
 name: Ascend NPU Test Suite
 
 on:
@@ -36,7 +33,8 @@ on:
           - ascendai/cann:7.1-openeuler2203sp2
           - ascendai/cann:8.0.rc2.alpha003-910b-ubuntu22.04-py3.9
           - ascendai/cann:8.0.rc3.alpha002-910b-ubuntu22.04-py3.9
-        default: 'ascendai/cann:8.0.rc3.alpha002-910b-ubuntu22.04-py3.9'
+          - latest
+        default: 'latest'
         description: 'The docker image which will be loaded'
       device:
         required: true
@@ -73,52 +71,37 @@ jobs:
           set -e
           echo "runner=${{ github.event.inputs.runner || 'self-hosted' }}" >> $GITHUB_OUTPUT
           echo "device=${{ github.event.inputs.device || '/dev/davinci6' }}" >> $GITHUB_OUTPUT
-          echo "image=${{ github.event.inputs.image || 'ascendai/cann:8.0.rc3.alpha002-910b-ubuntu22.04-py3.9' }}" >> $GITHUB_OUTPUT
-
-  # TODO(shink): remove this job, use actions/checkout instead
-  fetch-and-rebase:
-    name: Fetch and rebase
-    runs-on: ${{ needs.prepare.outputs.runner }}
-    needs:
-      - prepare
-    steps:
-      - name: Pull latest codes for torch_npu
-        uses: ./.github/actions/fetch-and-rebase
-        with:
-          repo_path: /home/runner/actions-runner/codes/npu/pytorch
-          remote_branch: upstream/master
-          loop: 10
+          echo "image=${{ github.event.inputs.image || 'latest' }}" >> $GITHUB_OUTPUT
 
   build:
     name: Build torch_npu
     needs:
       - prepare
-      - fetch-and-rebase
     uses: ./.github/workflows/_ascend_npu_build.yml
     with:
       runner: ${{ needs.prepare.outputs.runner }}
       image: ${{ needs.prepare.outputs.image }}
 
-  test:
-    name: Test torch_npu
-    needs:
-      - prepare
-      - build
-    uses: ./.github/workflows/_ascend_npu_test.yml
-    with:
-      runner: ${{ needs.prepare.outputs.runner }}
-      image: ${{ needs.prepare.outputs.image }}
-      device: ${{ needs.prepare.outputs.device }}
-      artifact_name: ${{ needs.build.outputs.artifact_name }}
-
-  pytorch-examples:
-    name: Run models
-    needs:
-      - prepare
-      - test
-    uses: ./.github/workflows/_ascend_npu_run_pytorch_examples.yml
-    with:
-      runner: ${{ needs.prepare.outputs.runner }}
-      image: ${{ needs.prepare.outputs.image }}
-      device: ${{ needs.prepare.outputs.device }}
-      artifact_name: ${{ needs.build.outputs.artifact_name }}
+#  test:
+#    name: Test torch_npu
+#    needs:
+#      - prepare
+#      - build
+#    uses: ./.github/workflows/_ascend_npu_test.yml
+#    with:
+#      runner: ${{ needs.prepare.outputs.runner }}
+#      image: ${{ needs.prepare.outputs.image }}
+#      device: ${{ needs.prepare.outputs.device }}
+#      artifact_name: ${{ needs.build.outputs.artifact_name }}
+#
+#  pytorch-examples:
+#    name: Run models
+#    needs:
+#      - prepare
+#      - test
+#    uses: ./.github/workflows/_ascend_npu_run_pytorch_examples.yml
+#    with:
+#      runner: ${{ needs.prepare.outputs.runner }}
+#      image: ${{ needs.prepare.outputs.image }}
+#      device: ${{ needs.prepare.outputs.device }}
+#      artifact_name: ${{ needs.build.outputs.artifact_name }}