From 30d98611a3a35287c47ded9647f0b4c81fbdf036 Mon Sep 17 00:00:00 2001 From: chuanqiw Date: Thu, 30 May 2024 12:10:12 +0000 Subject: [PATCH] [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 --- .github/workflows/periodic.yml | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/.github/workflows/periodic.yml b/.github/workflows/periodic.yml index 716a72cc6d2..cf2da41c1c4 100644 --- a/.github/workflows/periodic.yml +++ b/.github/workflows/periodic.yml @@ -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 }}