Skip to content

Conversation

@charlifu
Copy link

We have padding weights enabled by default for moe, but the weights are not padded by default in the unit test for moe.

This PR adds padded weights into the unit test to reflect the change of our moe kernel with padding.

@charlifu charlifu requested a review from gshtras August 12, 2024 16:45
@charlifu charlifu merged commit f49dff3 into main Aug 12, 2024
@gshtras gshtras deleted the charlifu/fix_moe_test_for_padding branch August 20, 2024 18:34
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