Skip to content

EmbeddingDenseBackward: Remove padding_idx cast to double #9406

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

Open
wants to merge 3 commits into
base: master
Choose a base branch
from

Conversation

unterumarmung
Copy link

Fixes #9392

@unterumarmung unterumarmung force-pushed the embedding-cast-remove branch from b3025d5 to 6f68495 Compare June 25, 2025 11:06
@unterumarmung unterumarmung changed the title EmbeddingDenseBackward: Remove padding_idx double cast EmbeddingDenseBackward: Remove padding_idx cast to double Jun 25, 2025
@bhavya01
Copy link
Collaborator

Thanks for the change! I am going to run the CI and see if it breaks anything

@bhavya01 bhavya01 self-requested a review June 25, 2025 16:50
@bhavya01
Copy link
Collaborator

Can you also please add a test for embedding backward similar to this one

def test_rrelu_module(self):

@unterumarmung
Copy link
Author

unterumarmung commented Jun 26, 2025

Done.

Please, rerun the CI

@unterumarmung unterumarmung changed the title EmbeddingDenseBackward: Remove padding_idx cast to double EmbeddingDenseBackward: Remove padding_idx cast to double Jun 26, 2025
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.

Unnecessary FP64 cast for padding_idx in EmbeddingDenseBackward
2 participants