Skip to content

[Kernel] Add back batch size 1536 and 3072 to MoE tuning#5242

Merged
WoosukKwon merged 1 commit intomainfrom
missed-batch-sizes
Jun 4, 2024
Merged

[Kernel] Add back batch size 1536 and 3072 to MoE tuning#5242
WoosukKwon merged 1 commit intomainfrom
missed-batch-sizes

Conversation

@WoosukKwon
Copy link
Copy Markdown
Collaborator

This PR adds back the batch size 1536 and 3072 in moe tuning. The batch sizes were mistakenly omitted during the refactoring #4921.

@WoosukKwon WoosukKwon marked this pull request as ready for review June 4, 2024 08:07
@WoosukKwon WoosukKwon requested a review from pcmoritz June 4, 2024 08:13
Copy link
Copy Markdown
Collaborator

@pcmoritz pcmoritz left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for the fix!

@WoosukKwon WoosukKwon merged commit 27208be into main Jun 4, 2024
@WoosukKwon WoosukKwon deleted the missed-batch-sizes branch June 4, 2024 16:58
chengzhi-lu pushed a commit to chengzhi-lu/vllm-infersche that referenced this pull request Jun 6, 2024
xjpang pushed a commit to xjpang/vllm that referenced this pull request Jun 27, 2024
xjpang pushed a commit to xjpang/vllm that referenced this pull request Jul 8, 2024
xjpang pushed a commit to xjpang/vllm that referenced this pull request Jul 24, 2024
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