Skip to content

Commit f5bbe4a

Browse files
committed
[CI] Enable setup xpu in linux job v2 workflow
1 parent 6a9e302 commit f5bbe4a

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

.github/workflows/linux_job_v2.yml

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,12 +176,16 @@ jobs:
176176

177177
- name: Setup Linux
178178
uses: ./test-infra/.github/actions/setup-linux
179-
if: ${{ inputs.gpu-arch-type != 'rocm' }}
179+
if: ${{ inputs.gpu-arch-type != 'rocm' || inputs.gpu-arch-version != 'xpu' }}
180180

181181
- name: Setup ROCM
182182
uses: pytorch/pytorch/.github/actions/setup-rocm@main
183183
if: ${{ inputs.gpu-arch-type == 'rocm' }}
184184

185+
- name: Setup XPU
186+
uses: pytorch/pytorch/.github/actions/setup-xpu@main
187+
if: ${{ inputs.gpu-arch-type == 'xpu' }}
188+
185189
- name: Setup SSH
186190
uses: ./test-infra/.github/actions/setup-ssh
187191
with:

0 commit comments

Comments
 (0)