Skip to content

fix gqa rotary dim 1#19874

Merged
aciddelgado merged 1 commit intomainfrom
aciddelgado/fix_rot_gqa
Mar 13, 2024
Merged

fix gqa rotary dim 1#19874
aciddelgado merged 1 commit intomainfrom
aciddelgado/fix_rot_gqa

Conversation

@aciddelgado
Copy link
Contributor

Description

GQA Rotary Dimension 1 incorrectly assumed to be based on head size.

Motivation and Context

This change should enable us to run phi-2 with GQA and Rotary Embedding fused.

@aciddelgado aciddelgado merged commit 8eb49c5 into main Mar 13, 2024
@aciddelgado aciddelgado deleted the aciddelgado/fix_rot_gqa branch March 13, 2024 21:09
YUNQIUGUO pushed a commit that referenced this pull request Mar 21, 2024
### Description
GQA Rotary Dimension 1 incorrectly assumed to be based on head size.



### Motivation and Context
This change should enable us to run phi-2 with GQA and Rotary Embedding
fused.
mszhanyi pushed a commit that referenced this pull request Apr 9, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants