wewe

Author: cllouud

.github/workflows/ascend_npu_test.yml

@@ -79,94 +79,18 @@ on:
 #
 # Doc: https://docs.github.com/en/actions/writing-workflows/workflow-syntax-for-github-actions#concurrency
 # Discussion: https://github.com/orgs/community/discussions/5435
-concurrency:
-  group: ${{ github.workflow }}-${{ github.event_name }}-${{ github.event.client_payload.pull_request.number || github.ref }}
-  cancel-in-progress: ${{ github.event_name == 'pull_request' }}
 
 jobs:
-  prepare:
-    name: Prepare
-    runs-on: ubuntu-latest
-    outputs:
-      runner: ${{ steps.set-env.outputs.runner }}
-      image: ${{ steps.set-env.outputs.image }}
-      device: ${{ steps.set-env.outputs.device }}
-      ref: ${{ steps.list-ref.outputs.ref }}
-      pr-number: ${{ steps.list-ref.outputs.pr_number }}
+  job_0:
+    runs-on: linux-arm64-npu-1
+    container:
+      image: ascendai/cann:latest
+      
     steps:
-      - name: Set environment params
-        id: set-env
+      - name: Show NPU info
         run: |
-          echo "runner=${{ github.event.inputs.runner || 'linux-arm64-npu-1' }}" >> $GITHUB_OUTPUT
-          echo "device=${{ github.event.inputs.device || '/dev/davinci5' }}" >> $GITHUB_OUTPUT
-          echo "image=${{ github.event.inputs.image || 'ascendai/cann:latest' }}" >> $GITHUB_OUTPUT
+          npu-smi info
 
-      # 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 "pr_number=${{ github.event.client_payload.pull_request.number }}" >> $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 }}
-      image: ${{ needs.prepare.outputs.image }}
-      ref: ${{ needs.prepare.outputs.ref }}
-      pr-number: ${{ needs.prepare.outputs.pr-number }}
-
-  build:
-    name: Build torch_npu
-    needs:
-      - prepare
-      - build-torch
-    if: ${{ !cancelled() && (success() || needs.build-torch.result == 'skipped') }}
-    uses: ./.github/workflows/_ascend_npu_build_torch_npu.yml
-    with:
-      runner: ${{ needs.prepare.outputs.runner }}
-      image: ${{ needs.prepare.outputs.image }}
-      device: ${{ needs.prepare.outputs.device }}
-      torch-artifact: ${{ needs.build-torch.outputs.torch-artifact }}
-
-  test:
-    name: Test torch_npu
-    needs:
-      - prepare
-      - build-torch
-      - build
-    if: |
-      !cancelled() && github.event_name != 'repository_dispatch' &&
-      (success() || (needs.build-torch.result == 'skipped' && needs.build.result == 'success'))
-    uses: ./.github/workflows/_ascend_npu_ut.yml
-    with:
-      runner: ${{ needs.prepare.outputs.runner }}
-      image: ${{ needs.prepare.outputs.image }}
-      device: ${{ needs.prepare.outputs.device }}
-      torch-artifact: ${{ needs.build-torch.outputs.torch-artifact }}
-      torch-npu-artifact: ${{ needs.build.outputs.torch-npu-artifact }}
-
-  benchmark:
-    name: Run benchmarks
-    needs:
-      - prepare
-      - build-torch
-      - build
-      - test
-    if: |
-      !cancelled() && github.event_name != 'repository_dispatch' &&
-      (success() || (needs.build-torch.result == 'skipped' && needs.build.result == 'success'))
-    uses: ./.github/workflows/_ascend_npu_benchmark.yml
-    with:
-      runner: ${{ needs.prepare.outputs.runner }}
-      image: ${{ needs.prepare.outputs.image }}
-      device: ${{ needs.prepare.outputs.device }}
-      torch-artifact: ${{ needs.build-torch.outputs.torch-artifact }}
-      torch-npu-artifact: ${{ needs.build.outputs.torch-npu-artifact }}
-    secrets:
-      pr-token: ${{ secrets.COSDT_BOT_TOKEN }}
+      - name: Sleep
+        run: sleep 3600
+