Skip to content

build: pin Megatron-LM odd top-k fallback - #20

Merged
pstefa1707 merged 2 commits into
trainers-mainfrom
jackrao/fix-cudnn-odd-topk
Jul 17, 2026
Merged

build: pin Megatron-LM odd top-k fallback#20
pstefa1707 merged 2 commits into
trainers-mainfrom
jackrao/fix-cudnn-odd-topk

Conversation

@JackRao123

Copy link
Copy Markdown

Summary

Test plan

  • Megatron-LM focused tests pass on the B200 devbox
  • Patched [235, 705] odd top-K primitive passes on B200

Made with Cursor

JackRao123 and others added 2 commits July 16, 2026 18:55
Advance Megatron-LM to the cuDNN DSA odd top-k fallback so packed CP segments avoid the frontend vector-width assertion.

Co-authored-by: Cursor <cursoragent@cursor.com>
@pstefa1707
pstefa1707 merged commit f7be4c1 into trainers-main Jul 17, 2026
4 checks passed
@pstefa1707
pstefa1707 deleted the jackrao/fix-cudnn-odd-topk branch July 17, 2026 03:00
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