Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 0 additions & 11 deletions .github/workflows/_ascend_npu_benchmark.yml
Original file line number Diff line number Diff line change
Expand Up @@ -11,10 +11,6 @@ on:
required: true
type: string
description: "The docker image which will be loaded"
device:
required: true
type: string
description: "The device selected to run on"
torch-artifact:
required: false
type: string
Expand All @@ -28,13 +24,6 @@ on:
description: "A token used to create a pull request"
required: true

# Bash shells do not use ~/.profile or ~/.bashrc so these shells need to be explicitly
# declared as "shell: bash -el {0}" on steps that need to be properly activated.
# It's used to activate ascend-toolkit environment variables.
defaults:
run:
shell: bash -el {0}

jobs:
benchmark:
name: run benchmarks for torch_npu
Expand Down
7 changes: 0 additions & 7 deletions .github/workflows/_ascend_npu_build_torch.yml
Original file line number Diff line number Diff line change
Expand Up @@ -26,13 +26,6 @@ on:
description: "The distribution artifact name of torch"
value: ${{ jobs.build.outputs.dist-name }}

# Bash shells do not use ~/.profile or ~/.bashrc so these shells need to be explicitly
# declared as "shell: bash -el {0}" on steps that need to be properly activated.
# It's used to activate ascend-toolkit environment variables.
defaults:
run:
shell: bash -el {0}

jobs:
build:
name: build torch for ${{ inputs.pr-number && format('#{0}', inputs.pr-number) || inputs.ref }}
Expand Down
11 changes: 0 additions & 11 deletions .github/workflows/_ascend_npu_build_torch_npu.yml
Original file line number Diff line number Diff line change
Expand Up @@ -11,10 +11,6 @@ on:
required: true
type: string
description: "The docker image which will be used to build"
device:
required: true
type: string
description: "The device selected to run on"
torch-artifact:
required: false
type: string
Expand All @@ -28,13 +24,6 @@ on:
description: "A token used to pull private repo"
required: true

# Bash shells do not use ~/.profile or ~/.bashrc so these shells need to be explicitly
# declared as "shell: bash -el {0}" on steps that need to be properly activated.
# It's used to activate ascend-toolkit environment variables.
defaults:
run:
shell: bash -el {0}

jobs:
build:
name: build torch_npu
Expand Down
11 changes: 0 additions & 11 deletions .github/workflows/_ascend_npu_ut.yml
Original file line number Diff line number Diff line change
Expand Up @@ -11,10 +11,6 @@ on:
required: true
type: string
description: "The docker image which will be loaded"
device:
required: true
type: string
description: "The device selected to run on"
torch-artifact:
required: false
type: string
Expand All @@ -24,13 +20,6 @@ on:
type: string
description: "The distribution artifact name of torch_npu"

# Bash shells do not use ~/.profile or ~/.bashrc so these shells need to be explicitly
# declared as "shell: bash -el {0}" on steps that need to be properly activated.
# It's used to activate ascend-toolkit environment variables.
defaults:
run:
shell: bash -el {0}

jobs:
test:
name: test torch_npu
Expand Down
22 changes: 1 addition & 21 deletions .github/workflows/ascend_npu_test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -56,20 +56,6 @@ on:
- ascendai/cann:latest
default: "ascendai/cann:latest"
description: "The docker image which will be loaded"
device:
required: true
type: choice
options:
- /dev/davinci1
- /dev/davinci2
- /dev/davinci3
- /dev/davinci4
- /dev/davinci5
- /dev/davinci6
- /dev/davinci7
- /dev/davinci8
default: "/dev/davinci5"
description: "The device selected to run on"

# Only cancel the previous runs when triggered by a pull_request event
#
Expand All @@ -90,15 +76,13 @@ jobs:
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 }}
steps:
- name: Set environment params
id: set-env
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

# TODO(shink): List ghstack PR's ref
Expand Down Expand Up @@ -131,10 +115,9 @@ jobs:
with:
runner: ${{ needs.prepare.outputs.runner }}
image: ${{ needs.prepare.outputs.image }}
device: ${{ needs.prepare.outputs.device }}
torch-artifact: ${{ needs.build-torch.outputs.torch-artifact }}
secrets:
runner-token: ${{ secrets.ASCEND_RUNNER_TOKEN }}
runner-token: ${{ secrets.ASCEND_RUNNER_TOKEN }}

test:
name: Test torch_npu
Expand All @@ -149,7 +132,6 @@ jobs:
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 }}

Expand All @@ -159,15 +141,13 @@ jobs:
- 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:
Expand Down
Loading