File tree Expand file tree Collapse file tree 2 files changed +1
-23
lines changed Expand file tree Collapse file tree 2 files changed +1
-23
lines changed Original file line number Diff line number Diff line change @@ -38,7 +38,7 @@ defaults:
38
38
39
39
jobs :
40
40
setup_environment :
41
- name : Run Torchtitan Environment
41
+ name : run torchtitan tests
42
42
runs-on : ${{ inputs.runner }}
43
43
container :
44
44
image : ${{ inputs.image }}
@@ -112,10 +112,6 @@ jobs:
112
112
run : |
113
113
pip install ${{ inputs.torch-npu-artifact }}
114
114
115
- - name : Install nightly torchvision and torchaudio
116
- run : |
117
- pip install --pre torchvision torchaudio --no-deps --index-url https://download.pytorch.org/whl/nightly/cpu
118
-
119
115
- name : Install project dependencies
120
116
run : |
121
117
pip install -r requirements.txt
@@ -132,7 +128,6 @@ jobs:
132
128
- name : Run torchtitan integration_test
133
129
working-directory : torchtitan
134
130
run : |
135
- mkdir artifacts-to-be-uploaded
136
131
npu_count=$(python -c "import torch; print(torch.npu.device_count())")
137
132
python ./tests/integration_tests.py artifacts-to-be-uploaded --ngpu ${npu_count}
138
133
Original file line number Diff line number Diff line change @@ -136,23 +136,6 @@ jobs:
136
136
device : ${{ needs.prepare.outputs.device }}
137
137
torch-artifact : ${{ needs.build-torch.outputs.torch-artifact }}
138
138
139
- test :
140
- name : Test torch_npu
141
- needs :
142
- - prepare
143
- - build-torch
144
- - build
145
- if : |
146
- !cancelled() && github.event_name != 'repository_dispatch' &&
147
- (success() || (needs.build-torch.result == 'skipped' && needs.build.result == 'success'))
148
- uses : ./.github/workflows/_ascend_npu_ut.yml
149
- with :
150
- runner : ${{ needs.prepare.outputs.runner }}
151
- image : ${{ needs.prepare.outputs.image }}
152
- device : ${{ needs.prepare.outputs.device }}
153
- torch-artifact : ${{ needs.build-torch.outputs.torch-artifact }}
154
- torch-npu-artifact : ${{ needs.build.outputs.torch-npu-artifact }}
155
-
156
139
torchtitan :
157
140
name : Run torchtitan
158
141
needs :
You can’t perform that action at this time.
0 commit comments