|
10 | 10 | - ".github/workflows/_ascend_npu_build_torch_npu.yml" |
11 | 11 | - ".github/workflows/_ascend_npu_ut.yml" |
12 | 12 | - ".github/workflows/_ascend_npu_benchmark.yml" |
13 | | - - ".github/workflows/_ascend_npu_torchtitan.yml" |
14 | 13 | - ".ci/**" |
15 | 14 | - "ascend_npu/**" |
16 | 15 | - "src/**" |
|
24 | 23 | - ".github/workflows/_ascend_npu_build_torch_npu.yml" |
25 | 24 | - ".github/workflows/_ascend_npu_ut.yml" |
26 | 25 | - ".github/workflows/_ascend_npu_benchmark.yml" |
27 | | - - ".github/workflows/_ascend_npu_torchtitan.yml" |
28 | 26 | - ".ci/**" |
29 | 27 | - "ascend_npu/**" |
30 | 28 | - "src/**" |
@@ -120,19 +118,36 @@ jobs: |
120 | 118 | image: ${{ needs.prepare.outputs.image }} |
121 | 119 | torch-artifact: ${{ needs.build-torch.outputs.torch-artifact }} |
122 | 120 |
|
123 | | - torchtitan: |
124 | | - name: Run torchtitan |
| 121 | + test: |
| 122 | + name: Test torch_npu |
125 | 123 | needs: |
126 | 124 | - prepare |
127 | 125 | - build-torch |
128 | 126 | - build |
129 | 127 | if: | |
130 | 128 | !cancelled() && github.event_name != 'repository_dispatch' && |
131 | 129 | (success() || (needs.build-torch.result == 'skipped' && needs.build.result == 'success')) |
132 | | - uses: ./.github/workflows/_ascend_npu_torchtitan.yml |
| 130 | + uses: ./.github/workflows/_ascend_npu_ut.yml |
133 | 131 | with: |
134 | 132 | runner: ${{ needs.prepare.outputs.runner }} |
135 | 133 | image: ${{ needs.prepare.outputs.image }} |
136 | 134 | torch-artifact: ${{ needs.build-torch.outputs.torch-artifact }} |
137 | 135 | torch-npu-artifact: ${{ needs.build.outputs.torch-npu-artifact }} |
138 | 136 |
|
| 137 | + benchmark: |
| 138 | + name: Run benchmarks |
| 139 | + needs: |
| 140 | + - prepare |
| 141 | + - build-torch |
| 142 | + - build |
| 143 | + if: | |
| 144 | + !cancelled() && github.event_name != 'repository_dispatch' && |
| 145 | + (success() || (needs.build-torch.result == 'skipped' && needs.build.result == 'success')) |
| 146 | + uses: ./.github/workflows/_ascend_npu_benchmark.yml |
| 147 | + with: |
| 148 | + runner: ${{ needs.prepare.outputs.runner }} |
| 149 | + image: ${{ needs.prepare.outputs.image }} |
| 150 | + torch-artifact: ${{ needs.build-torch.outputs.torch-artifact }} |
| 151 | + torch-npu-artifact: ${{ needs.build.outputs.torch-npu-artifact }} |
| 152 | + secrets: |
| 153 | + pr-token: ${{ secrets.COSDT_BOT_TOKEN }} |
0 commit comments