5656 - ascendai/cann:latest
5757 default : " ascendai/cann:latest"
5858 description : " The docker image which will be loaded"
59- device :
60- required : true
61- type : choice
62- options :
63- - /dev/davinci1
64- - /dev/davinci2
65- - /dev/davinci3
66- - /dev/davinci4
67- - /dev/davinci5
68- - /dev/davinci6
69- - /dev/davinci7
70- - /dev/davinci8
71- default : " /dev/davinci5"
72- description : " The device selected to run on"
7359
7460# Only cancel the previous runs when triggered by a pull_request event
7561#
9884 id : set-env
9985 run : |
10086 echo "runner=${{ github.event.inputs.runner || 'linux-arm64-npu-1' }}" >> $GITHUB_OUTPUT
101- echo "device=${{ github.event.inputs.device || '/dev/davinci5' }}" >> $GITHUB_OUTPUT
10287 echo "image=${{ github.event.inputs.image || 'ascendai/cann:latest' }}" >> $GITHUB_OUTPUT
10388
10489 # TODO(shink): List ghstack PR's ref
@@ -131,7 +116,6 @@ jobs:
131116 with :
132117 runner : ${{ needs.prepare.outputs.runner }}
133118 image : ${{ needs.prepare.outputs.image }}
134- device : ${{ needs.prepare.outputs.device }}
135119 torch-artifact : ${{ needs.build-torch.outputs.torch-artifact }}
136120
137121 test :
@@ -147,7 +131,6 @@ jobs:
147131 with :
148132 runner : ${{ needs.prepare.outputs.runner }}
149133 image : ${{ needs.prepare.outputs.image }}
150- device : ${{ needs.prepare.outputs.device }}
151134 torch-artifact : ${{ needs.build-torch.outputs.torch-artifact }}
152135 torch-npu-artifact : ${{ needs.build.outputs.torch-npu-artifact }}
153136
@@ -157,15 +140,13 @@ jobs:
157140 - prepare
158141 - build-torch
159142 - build
160- - test
161143 if : |
162144 !cancelled() && github.event_name != 'repository_dispatch' &&
163145 (success() || (needs.build-torch.result == 'skipped' && needs.build.result == 'success'))
164146 uses : ./.github/workflows/_ascend_npu_benchmark.yml
165147 with :
166148 runner : ${{ needs.prepare.outputs.runner }}
167149 image : ${{ needs.prepare.outputs.image }}
168- device : ${{ needs.prepare.outputs.device }}
169150 torch-artifact : ${{ needs.build-torch.outputs.torch-artifact }}
170151 torch-npu-artifact : ${{ needs.build.outputs.torch-npu-artifact }}
171152 secrets :
0 commit comments