Skip to content

Allow int data-type for Embedding indices.#6718

Merged
ysiraichi merged 3 commits intomasterfrom
ysiraichi/fix-embedding-indices-type
Mar 12, 2024
Merged

Allow int data-type for Embedding indices.#6718
ysiraichi merged 3 commits intomasterfrom
ysiraichi/fix-embedding-indices-type

Conversation

@ysiraichi
Copy link
Copy Markdown
Collaborator

Fix: #6648

This PR allows Int indices on Embedding lowering, instead of strictly checking for Long only. This is allowed according to the documentation.

cc @miladm @JackCaoG @bhavya01 @lezcano

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[torchbench] llama failing when executed with bfloat16 data-type.

3 participants