@@ -124,7 +124,7 @@ jobs:
124
124
echo "::endgroup::"
125
125
126
126
echo "::group::Install required packages"
127
- pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121
127
+ pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121
128
128
pip install -r ./requirements.txt
129
129
pip list
130
130
echo "::endgroup::"
@@ -140,37 +140,5 @@ jobs:
140
140
141
141
echo "::group::Run inference"
142
142
bash .ci/scripts/validate.sh "./checkpoints/${REPO_NAME}/model.pth" "cuda" "compile"
143
- test-gpu-aoti :
144
- uses : pytorch/test-infra/.github/workflows/linux_job.yml@main
145
- name : test-gpu-aoti (${{ matrix.platform }}, ${{ matrix.model_name }})
146
- needs : gather-models-gpu
147
- strategy :
148
- matrix : ${{ fromJSON(needs.gather-models-gpu.outputs.models) }}
149
- fail-fast : false
150
- with :
151
- runner : linux.g5.4xlarge.nvidia.gpu
152
- gpu-arch-type : cuda
153
- gpu-arch-version : " 12.1"
154
- script : |
155
- echo "::group::Print machine info"
156
- nvidia-smi
157
- echo "::endgroup::"
158
-
159
- echo "::group::Install required packages"
160
- pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121
161
- pip install -r ./requirements.txt
162
- pip list
163
- echo "::endgroup::"
164
-
165
- echo "::group::Download checkpoint"
166
- export REPO_NAME=${{ matrix.repo_name }}
167
- bash .ci/scripts/wget_checkpoint.sh ${REPO_NAME} ${{ matrix.resources }}
168
- echo "::endgroup::"
169
-
170
- echo "::group::Convert checkpoint"
171
- bash .ci/scripts/convert_checkpoint.sh ${REPO_NAME}
172
- echo "::endgroup::"
173
-
174
- echo "::group::Run inference"
175
143
bash .ci/scripts/validate.sh "./checkpoints/${REPO_NAME}/model.pth" "cuda" "aoti"
176
144
echo "::endgroup::"
0 commit comments