[CI] add xpu test in periodic workflow (#126410)

Works for https://github.com/pytorch/pytorch/issues/114850
Pull Request resolved: https://github.com/pytorch/pytorch/pull/126410
Approved by: https://github.com/EikanWang, https://github.com/atalman
This commit is contained in:
chuanqiw
2024-05-30 12:10:12 +00:00
committed by PyTorch MergeBot
parent 1071437169
commit 30d98611a3

View File

@@ -242,3 +242,30 @@ jobs:
build-environment: linux-focal-rocm6.1-py3.8
docker-image: ${{ needs.linux-focal-rocm6_1-py3_8-build.outputs.docker-image }}
test-matrix: ${{ needs.linux-focal-rocm6_1-py3_8-build.outputs.test-matrix }}
linux-jammy-xpu-py3_8-build:
name: linux-jammy-xpu-py3.8
uses: ./.github/workflows/_linux-build.yml
with:
build-environment: linux-jammy-xpu-py3.8
docker-image-name: pytorch-linux-jammy-xpu-2024.0-py3
runner: linux.2xlarge
test-matrix: |
{ include: [
{ config: "default", shard: 1, num_shards: 4, runner: "linux.idc.xpu" },
{ config: "default", shard: 2, num_shards: 4, runner: "linux.idc.xpu" },
{ config: "default", shard: 3, num_shards: 4, runner: "linux.idc.xpu" },
{ config: "default", shard: 4, num_shards: 4, runner: "linux.idc.xpu" },
]}
linux-jammy-xpu-py3_8-test:
name: linux-jammy-xpu-py3.8
uses: ./.github/workflows/_xpu-test.yml
needs: linux-jammy-xpu-py3_8-build
permissions:
id-token: write
contents: read
with:
build-environment: linux-jammy-xpu-py3.8
docker-image: ${{ needs.linux-jammy-xpu-py3_8-build.outputs.docker-image }}
test-matrix: ${{ needs.linux-jammy-xpu-py3_8-build.outputs.test-matrix }}