There was an error while loading. Please reload this page.
1 parent 6a9e302 commit f5bbe4aCopy full SHA for f5bbe4a
1 file changed
.github/workflows/linux_job_v2.yml
@@ -176,12 +176,16 @@ jobs:
176
177
- name: Setup Linux
178
uses: ./test-infra/.github/actions/setup-linux
179
- if: ${{ inputs.gpu-arch-type != 'rocm' }}
+ if: ${{ inputs.gpu-arch-type != 'rocm' || inputs.gpu-arch-version != 'xpu' }}
180
181
- name: Setup ROCM
182
uses: pytorch/pytorch/.github/actions/setup-rocm@main
183
if: ${{ inputs.gpu-arch-type == 'rocm' }}
184
185
+ - name: Setup XPU
186
+ uses: pytorch/pytorch/.github/actions/setup-xpu@main
187
+ if: ${{ inputs.gpu-arch-type == 'xpu' }}
188
+
189
- name: Setup SSH
190
uses: ./test-infra/.github/actions/setup-ssh
191
with:
0 commit comments