mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
bcc98bb2a4
This unblocks https://github.com/pytorch/test-infra/issues/6869. The key changes to call out: * B200 needs OIDC to access ECR and upload stats to S3, so we need to set `id-token: write` in `_linux-test`. All workflows calling `_linux-test` also need to be updated accordingly * Connecting sccache to S3 on B200 doesn't seem to work, so I disable it. It still works locally though. ### Testing https://github.com/pytorch/pytorch/actions/runs/16055549292/job/45312298376 Pull Request resolved: https://github.com/pytorch/pytorch/pull/157341 Approved by: https://github.com/nWEIdia, https://github.com/atalman, https://github.com/malfet
78 lines
2.8 KiB
YAML
78 lines
2.8 KiB
YAML
name: s390x-periodic
|
|
|
|
on:
|
|
schedule:
|
|
# We have several schedules so jobs can check github.event.schedule to activate only for a fraction of the runs.
|
|
# Also run less frequently on weekends.
|
|
- cron: 29 8 * * * # about 1:29am PDT, for mem leak check and rerun disabled tests
|
|
push:
|
|
tags:
|
|
- ciflow/periodic/*
|
|
- ciflow/s390/*
|
|
workflow_dispatch:
|
|
|
|
concurrency:
|
|
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-${{ github.ref_type == 'branch' && github.sha }}-${{ github.event_name == 'workflow_dispatch' }}-${{ github.event_name == 'schedule' }}-${{ github.event.schedule }}
|
|
cancel-in-progress: true
|
|
|
|
permissions:
|
|
id-token: write
|
|
contents: read
|
|
|
|
jobs:
|
|
llm-td:
|
|
if: github.repository_owner == 'pytorch'
|
|
name: before-test
|
|
uses: ./.github/workflows/llm_td_retrieval.yml
|
|
permissions:
|
|
id-token: write
|
|
contents: read
|
|
|
|
target-determination:
|
|
name: before-test
|
|
uses: ./.github/workflows/target_determination.yml
|
|
needs: llm-td
|
|
permissions:
|
|
id-token: write
|
|
contents: read
|
|
|
|
linux-manylinux-2_28-py3-cpu-s390x-build:
|
|
if: github.repository_owner == 'pytorch'
|
|
name: linux-manylinux-2_28-py3-cpu-s390x
|
|
uses: ./.github/workflows/_linux-build.yml
|
|
with:
|
|
build-environment: linux-s390x-binary-manywheel
|
|
docker-image-name: pytorch/manylinuxs390x-builder:cpu-s390x
|
|
runner: linux.s390x
|
|
test-matrix: |
|
|
{ include: [
|
|
{ config: "default", shard: 1, num_shards: 10, runner: "linux.s390x" },
|
|
{ config: "default", shard: 2, num_shards: 10, runner: "linux.s390x" },
|
|
{ config: "default", shard: 3, num_shards: 10, runner: "linux.s390x" },
|
|
{ config: "default", shard: 4, num_shards: 10, runner: "linux.s390x" },
|
|
{ config: "default", shard: 5, num_shards: 10, runner: "linux.s390x" },
|
|
{ config: "default", shard: 6, num_shards: 10, runner: "linux.s390x" },
|
|
{ config: "default", shard: 7, num_shards: 10, runner: "linux.s390x" },
|
|
{ config: "default", shard: 8, num_shards: 10, runner: "linux.s390x" },
|
|
{ config: "default", shard: 9, num_shards: 10, runner: "linux.s390x" },
|
|
{ config: "default", shard: 10, num_shards: 10, runner: "linux.s390x" },
|
|
]}
|
|
secrets: inherit
|
|
|
|
linux-manylinux-2_28-py3-cpu-s390x-test:
|
|
permissions:
|
|
id-token: write
|
|
contents: read
|
|
name: linux-manylinux-2_28-py3-cpu-s390x
|
|
uses: ./.github/workflows/_linux-test.yml
|
|
needs:
|
|
- linux-manylinux-2_28-py3-cpu-s390x-build
|
|
- target-determination
|
|
with:
|
|
build-environment: linux-s390x-binary-manywheel
|
|
docker-image: pytorch/manylinuxs390x-builder:cpu-s390x
|
|
test-matrix: ${{ needs.linux-manylinux-2_28-py3-cpu-s390x-build.outputs.test-matrix }}
|
|
timeout-minutes: 600
|
|
use-gha: "yes"
|
|
secrets: inherit
|