transformers
Restore the `MegaBlocksMoeMLP` hub kernel mapping for MXFP4 GPT-OSS
#50150

Merged

Restore the `MegaBlocksMoeMLP` hub kernel mapping for MXFP4 GPT-OSS #50150

SunMarc merged 5 commits into huggingface:main from jiqing-feng:fix-mxfp4-megablocks-hub-mapping

Restore MegaBlocksMoeMLP hub kernel mapping for MXFP4 GPT-OSS MLP

d30ffdf7

Test the megablocks kernel on a real MXFP4 GPT-OSS model on CPU/XPU

182e197c

jiqing-feng marked this pull request as ready for review 8 days ago

Merge branch 'main' into fix-mxfp4-megablocks-hub-mapping

97aff097

github-actions requested a review from

ydshieh 8 days ago

github-actions requested a review from

SunMarc 8 days ago

Use the modeling flash-attn2 kernel version in benchmark_v2's FA2 check

2d28f92b

SunMarc commented on 2026-10-30

SunMarc approved these changes on 2026-10-30

Merge branch 'main' into fix-mxfp4-megablocks-hub-mapping

124a3fd5

SunMarc merged 955b6d35 into main 8 days ago

Reviewers

SunMarc

ydshieh

Assignees

No one assigned

Labels

None yet

Milestone

No milestone