Skip to content

Fixing the shape to use in padding calculation#464

Merged
gshtras merged 5 commits intomainfrom
moe_padding_shape_fix
Mar 7, 2025
Merged

Fixing the shape to use in padding calculation#464
gshtras merged 5 commits intomainfrom
moe_padding_shape_fix

Conversation

@gshtras
Copy link
Collaborator

@gshtras gshtras commented Mar 7, 2025

In preparation for upstreaming MoE padding
Fixing accuracy regression induced by vllm-project#12185
In upstream B.shape[2] and A.shape[1] are supposed to match, but not when using moe padding

shajrawi
shajrawi previously approved these changes Mar 7, 2025
divakar-amd
divakar-amd previously approved these changes Mar 7, 2025
@gshtras gshtras merged commit 3ee6551 into main Mar 7, 2025
2 of 4 checks passed
@gshtras gshtras deleted the moe_padding_shape_fix branch March 7, 2025 19:18
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.

3 participants