Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

bugfix: fix wrong padded_batch_size_ #296

Merged
merged 2 commits into from
Jun 11, 2024
Merged

bugfix: fix wrong padded_batch_size_ #296

merged 2 commits into from
Jun 11, 2024

Conversation

yzh119
Copy link
Collaborator

@yzh119 yzh119 commented Jun 11, 2024

In #294 , we set padded_batch_size_ to num_kv_heads * batch_size when no splitting kv, which should be batch_size

@yzh119 yzh119 merged commit aff4cf0 into main Jun 11, 2024
@yzh119 yzh119 deleted the bugfix-294 branch June 14, 2024 21: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.

1 participant