Skip to content

[CI] Bump triton to 3.8.0 - #5152

Open
yuyzhang512 wants to merge 1 commit into
ROCm:mainfrom
yuyzhang512:release-test/triton-release-3.8
Open

[CI] Bump triton to 3.8.0#5152
yuyzhang512 wants to merge 1 commit into
ROCm:mainfrom
yuyzhang512:release-test/triton-release-3.8

Conversation

@yuyzhang512

Copy link
Copy Markdown
Contributor

Motivation

Validate the coming Triton 3.8.0 release for AITER: point CI at the release_ wheel channel and bump the pinned Triton to 3.8.0. The gated-delta-rule block-pointer fix this depends on is already on main via #4950, so this PR only carries the CI changes.

Technical Details

  • .github/scripts/install_triton.sh and .github/scripts/download_triton_wheel.sh: switch the Triton wheel index channel to release_, so the index URL becomes https://pypi.amd.com/triton/release_/rocm-<ver>/simple/ (both the default and the rocm-core-derived paths), and bump the pinned Triton wheel from 3.7.0 to 3.8.0.
  • .github/workflows/triton-test.yaml (MI35X job): deselect op_tests/triton_tests/attention/test_mha_with_pe.py::test_mha_varlen_with_pe[True-0.17-96-64-8-1-64-128], which fails an accuracy check on gfx950 with the coming Triton release.
  • .github/workflows/triton-test.yaml (MI300X job): deselect all shapes of test_pa_prefill.py::test_contexted_kv_attention and test_chunked_pa_prefill.py::test_contexted_kv_attention. These fail an accuracy check on gfx942, and were reproduced locally hanging the GPU (HW Exception ... reason: GPU Hang) on Triton 3.8.0, so the whole node is deselected rather than individual shapes.

Test Plan

  • Confirm CI resolves the Triton 3.8.0 and triton-kernels wheels from the release_ index.
  • Confirm the MI35X and MI300X Triton shards pass with the deselected cases skipped.

Test Result

Submission Checklist

@yuyzhang512
yuyzhang512 requested a review from a team August 31, 2026 16:06
@github-actions

Copy link
Copy Markdown
Contributor

🏷️ CI Guide

Runs automatically on every PR:

  • ✅ Pre-checks (submodule verification, code formatting)
  • ✅ Aiter op tests (gfx942 + gfx950)
  • ✅ Triton tests on MI35X (only when aiter/ops/triton/** or related paths are changed)

Extended tests (opt-in via labels):

Label Tests
ci:gfx1250-ffm-triton Run the five-shard gfx1250 FFM Triton test suite
ci:triton-300x Run an additional Triton test job on MI300X in PRs; main branch always runs both MI35X and MI300X
multigpu Aiter multi-GPU tests on the 8-GPU runner
ci:sglang SGLang integration tests: DeepSeek-R1-MXFP4 accuracy, Qwen 3.5 accuracy
ci:atom ATOM benchmark: DeepSeek-R1-0528, GPT-OSS-120B
ci:atom_full ATOM accuracy suite for PR and main models from ATOM models_accuracy.json
ci:vllm vLLM benchmark: GPT-OSS-120B, DeepSeek-R1-0528, Kimi-K2.5
ci:all All standard extended tests (excludes ci:atom_full)

Only add ci:atom_full for FlyDSL or Triton upgrades.
Add labels via the sidebar or gh pr edit 5152 --add-label <label>

PR title tags & labels:
Component tags ([Triton/Gluon], [HIP], [CK], [ASM], ...) are added to the PR title and as PR labels automatically from the changed files and re-synced on every push — change-type tags like [fix]/[Perf], op tags like [MLA], and human labels (ci:*) are left untouched. Add the no-auto-title label to opt this PR out.

@zufayu
zufayu requested a review from gyohuangxin September 1, 2026 01:16
Point the Triton wheel index at the _release channel
(https://pypi.amd.com/triton/_release/rocm-<ver>/simple/) and bump the pinned
Triton wheel from 3.7.0 to 3.8.0 in install_triton.sh and
download_triton_wheel.sh.

MI35X: deselect test_mha_varlen_with_pe[True-0.17-96-64-8-1-64-128].
MI300X: deselect all shapes of test_pa_prefill and test_chunked_pa_prefill
test_contexted_kv_attention, which fail an accuracy check and can hang the GPU
with the coming Triton release.

The gated_delta_rule block-pointer fix is already on main via ROCm#4950.

Co-Authored-By: Claude <noreply@anthropic.com>
@yuyzhang512
yuyzhang512 force-pushed the release-test/triton-release-3.8 branch from b918e37 to cb9f2eb Compare September 1, 2026 02:22
@github-actions github-actions Bot changed the title Bump triton to 3.8.0 [CI] Bump triton to 3.8.0 Sep 1, 2026
@github-actions github-actions Bot added the CI label Sep 1, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant