pytorch
29fdb354 - [MPS] Fix embedding_backward() issue with Float16 (#94950)

Commit
1 year ago
[MPS] Fix embedding_backward() issue with Float16 (#94950) - Casting the float16 input tensor to float32 and cast back the output tensor Pull Request resolved: https://github.com/pytorch/pytorch/pull/94950 Approved by: https://github.com/DenisVieriu97
Author
Committer
Parents
Loading