Skip to content

Commit 19d1511

Browse files
authored
Update runs-on & device
1 parent 641a62b commit 19d1511

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

.github/workflows/deepspeed.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ on:
3434
- 'requirements/**'
3535
push:
3636
branches:
37-
- 'deepspeed'
37+
- 'main'
3838

3939
schedule:
4040
- cron: "0 0 * * *"
@@ -66,12 +66,12 @@ jobs:
6666
needs: prepare
6767
if: ${{ github.repository_owner == 'Ascend' }}
6868
name: Run unit tests with DeepSpeed
69-
runs-on: ${{ github.event.inputs.runner }}
69+
runs-on: ${{ needs.prepare.outputs.runner }}
7070
container:
7171
image: ascendai/cann:latest
7272
options: >-
7373
--network host
74-
--device ${{ github.event.inputs.device }}
74+
--device ${{ needs.prepare.outputs.device }}
7575
--device /dev/davinci_manager
7676
--device /dev/devmm_svm
7777
--device /dev/hisi_hdc

0 commit comments

Comments
 (0)