Skip to content

[AMD] Run the MiniMax indexer CP test on MI35X, not MI300 - #43259

Merged
hnyls2002 merged 2 commits into
sgl-project:mainfrom
kevin-mii:fix/minimax-indexer-cp-test-mi35x
Oct 9, 2026
Merged

hnyls2002 merged 2 commits into
sgl-project:mainfrom
kevin-mii:fix/minimax-indexer-cp-test-mi35x

Conversation

@kevin-mii

@kevin-mii kevin-mii commented Oct 9, 2026 •

Copy link
Copy Markdown
Collaborator

Motivation

test/registered/amd/test_minimax_indexer_cp.py (added in #41488) has failed on every stage-b-test-1-gpu-small-amd MI300 run since it landed, as @bingxche reported on #41488 (first failure, still failing 2026-10-08):

triton.runtime.errors.OutOfResources: out of resource: shared memory, Required: 131072, Hardware limit: 65536.

The error comes from the reference side of the test: flash_decode_with_topk_idx's score kernel needs 128 KB of LDS, and MI300 (gfx942) has 64 KB. Indexer CP itself is gfx950-only (unsupported_reason returns "requires AMD gfx950"), so the test belongs on MI35X anyway.

Modifications

Register the test in stage-b-test-1-gpu-small-amd-mi35x instead of stage-b-test-1-gpu-small-amd. No code change.

Accuracy Tests

On MI350X (gfx950):

python test/registered/amd/test_minimax_indexer_cp.py -f
Ran 1 test in 8.174s
OK

Suite collection: the file is now in stage-b-test-1-gpu-small-amd-mi35x and no longer in stage-b-test-1-gpu-small-amd; validate_all_suites passes.

Checklist

  • Format your code according to the Format code with pre-commit.
  • Add unit tests according to the Run and add unit tests. (moves an existing test)
  • Update documentation -- not applicable.
  • Provide accuracy and speed benchmark results -- not applicable.

🤖 Generated with Claude Code


CI States

Latest PR Test (Base): ❌ Run #37873098166
Latest PR Test (Extra): ❌ Run #37873097914
Latest PR Test (AMD ROCm 10): ❌ Run #37873098153

kevin-mii and others added 2 commits October 9, 2026 02:04
Indexer CP is gated to gfx950, and the native top-k selector the test
compares against needs 128 KB of LDS for its score kernel; MI300 (gfx942)
has 64 KB, so the test raised OutOfResources on every stage-b MI300 run
since it landed in sgl-project#41488. It passes on gfx950.

Reported by @bingxche on sgl-project#41488.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
The suite name already says where it runs; the reason is in the PR.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
@hnyls2002
hnyls2002 merged commit 70e82f7 into sgl-project:main Oct 9, 2026
109 of 121 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants