update

Author: shink

.github/workflows/_ascend_npu_benchmark.yml

@@ -39,7 +39,6 @@ jobs:
         - /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/
         - /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info
         - /etc/ascend_install.info:/etc/ascend_install.info
-        - /data/disk3/jyh/:/jyh
       options: >-
         --network host
         --device ${{ inputs.device }}
@@ -77,9 +76,10 @@ jobs:
           path: benchmark
 
       - name: Download ${{ inputs.artifact_name }}
-        working-directory: ascend_npu
-        run: |
-          cp /jyh/*.whl .
+        uses: actions/download-artifact@v4
+        with:
+          name: ${{ inputs.artifact_name }}
+          path: ascend_npu
 
       - name: Install torch_npu
         working-directory: ascend_npu

.github/workflows/ascend_npu_test.yml

@@ -73,34 +73,36 @@ jobs:
           echo "device=${{ github.event.inputs.device || '/dev/davinci5' }}" >> $GITHUB_OUTPUT
           echo "image=${{ github.event.inputs.image || 'ascendai/cann:latest' }}" >> $GITHUB_OUTPUT
 
-#  build:
-#    name: Build torch_npu
-#    needs:
-#      - prepare
-#    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 }}
+  build:
+    name: Build torch_npu
+    needs:
+      - prepare
+    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 }}
 
   benchmark:
     name: Run benchmarks
     needs:
       - prepare
+      - build
+      - test
     uses: ./.github/workflows/_ascend_npu_benchmark.yml
     with:
       runner: ${{ needs.prepare.outputs.runner }}
       image: ${{ needs.prepare.outputs.image }}
       device: ${{ needs.prepare.outputs.device }}
-      artifact_name: 'torch_npu-2.6.0+git69ceea7-cp310-cp310-linux_aarch64.whl'
+      artifact_name: ${{ needs.build.outputs.artifact_name }}