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

[DML EP] Support split hidden size for RotaryEmbedding #18852

Conversation

PatriceVignola
Copy link
Contributor

RotaryEmbedding now supports the [batchSize, numHeads, sequenceLength, headSize] format for its input, which is used in Mistral.

Copy link
Contributor

@sumitsays sumitsays left a comment

Choose a reason for hiding this comment

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

:shipit:

@PatriceVignola
Copy link
Contributor Author

/azp run Big Models

Copy link

Azure Pipelines successfully started running 1 pipeline(s).

@PatriceVignola PatriceVignola merged commit 61e07a4 into main Feb 13, 2024
97 of 98 checks passed
@PatriceVignola PatriceVignola deleted the user/pavignol/add-rotary-embedding-split-hidden-size-support branch February 13, 2024 03:36
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.

2 participants