Skip to content

[CI] Parity: replace ROCm nightly with scheduled Preview #3554

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
ethanwee1 wants to merge 3 commits into
base: develop
Choose a base branch
from
Open

[CI] Parity: replace ROCm nightly with scheduled Preview #3554

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
f3b83f4
[CI] Parity: replace ROCm nightly with scheduled Preview
ethanwee1 Aug 11, 2026
da3b0ad
[CI] Parity: keep Preview check-run matching resilient
ethanwee1 Aug 11, 2026
44b8c76
[CI] Parity: sync MI300 shard count
ethanwee1 Aug 11, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions .automation_scripts/pytorch-unit-test-scripts/download_testlogs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -526,7 +526,7 @@ def parse_args():
parser.add_argument('--no_rocm', action='store_true')
parser.add_argument('--no_cuda', action='store_true')
parser.add_argument('--pr_id', type=int, help='The pull request ID')
parser.add_argument('--arch', type=str, choices=['mi200', 'mi300', 'mi350', 'navi31', 'nightly'], default='mi350', help='ROCm GPU architecture (mi200, mi300, mi350, navi31, or nightly, default: mi350)')
parser.add_argument('--arch', type=str, choices=['mi200', 'mi300', 'mi350', 'navi31', 'preview'], default='mi350', help='ROCm GPU architecture (mi200, mi300, mi350, navi31, or preview, default: mi350)')
parser.add_argument('--include_inductor_periodic', action='store_true', help='Also download inductor-periodic benchmark artifacts (into a separate directory, not included in parity CSV)')
parser.add_argument('--baseline_sha', type=str, help='Baseline commit SHA to compare against. Downloads the same ROCm workflows for this commit into baseline_xml/.')
return parser.parse_args()
Expand Down Expand Up @@ -615,7 +615,7 @@ def main():
# in, job-name prefixes, shard counts and fallbacks) comes from
# parity_job_config.json - see PARITY_CONFIG at the top of this file.
global ROCmWorkflowNames
arch = args.arch # 'mi200', 'mi300', 'mi350', 'navi31', or 'nightly'
arch = args.arch # 'mi200', 'mi300', 'mi350', 'navi31', or 'preview'
arch_config = PARITY_CONFIG["rocm"][arch]

ROCmWorkflowNames, rocm_job_prefix, arch_fallbacks = parity_config_views(arch_config)
Expand Down
14 changes: 7 additions & 7 deletions .automation_scripts/pytorch-unit-test-scripts/parity_job_config.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@
"default": [{ "workflow": "rocm-mi300", "job_prefix": "linux-noble-rocm-py3.12-mi300" }],
"distributed": [{ "workflow": "periodic-rocm-mi300", "job_prefix": "linux-noble-rocm-py3.12-mi300" }],
"inductor": [{ "workflow": "inductor-rocm-mi300", "job_prefix": "linux-noble-rocm-py3.12-mi300" }],
"shard_counts": { "default": 6, "distributed": 3, "inductor": 2 },
"shard_counts": { "default": 8, "distributed": 3, "inductor": 2 },
"checkrun_regex": "rocm.*mi300.*/ test [(](default|distributed|inductor),"
},
"mi200": {
Expand All @@ -64,12 +64,12 @@
"shard_counts": { "default": 2, "distributed": 3, "inductor": 2 },
"checkrun_regex": "rocm.*navi31.*/ test [(]default,"
},
"nightly": {
"default": [{ "workflow": "rocm-nightly", "job_prefix": "linux-noble-rocm-nightly-py3.12-mi350" }],
"distributed": [{ "workflow": "rocm-nightly", "job_prefix": "linux-noble-rocm-nightly-py3.12-mi350" }],
"inductor": [{ "workflow": "rocm-nightly", "job_prefix": "linux-noble-rocm-nightly-py3.12-mi350" }],
"shard_counts": { "default": 6, "distributed": 3, "inductor": 2 },
"checkrun_regex": "rocm-nightly.*/ test [(](default|distributed|inductor),"
"preview": {
"default": [{ "workflow": "rocm-preview", "job_prefix": "linux-noble-rocm-preview-py3.12-mi350" }],
"distributed": [{ "workflow": "rocm-preview", "job_prefix": "linux-noble-rocm-preview-py3.12-mi350" }],
"inductor": [{ "workflow": "rocm-preview", "job_prefix": "linux-noble-rocm-preview-py3.12-mi350" }],
"shard_counts": { "default": 8, "distributed": 3, "inductor": 2 },
"checkrun_regex": "rocm-preview.*/ test [(](default|distributed|inductor),"
}
}
}
4 changes: 2 additions & 2 deletions .github/workflows/parity.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ on:
required: false
type: string
arch:
description: 'ROCm architectures, comma or space separated. Options: mi350, mi300, mi200, nightly, navi31. Example: "nightly, mi350" or "mi300"'
description: 'ROCm architectures, comma or space separated. Options: mi350, mi300, mi200, navi31, preview. Example: "preview, mi350" or "mi300"'
required: false
default: 'mi350, mi300, mi200'
type: string
Expand Down Expand Up @@ -72,7 +72,7 @@ on:
type: boolean
# summarize_xml_testreports flags
set1_name:
description: 'Label for ROCm columns in output CSV. Examples: rocm, nightly, mi300. Default: rocm'
description: 'Label for ROCm columns in output CSV. Examples: rocm, preview, mi300. Default: rocm'
required: false
default: 'rocm'
type: string
Expand Down