torch.multinomial
on MPS crashes with Error: total bytes of NDArray > 2**32'
#86279
Labels
module: mps
Related to Apple Metal Performance Shaders framework
module: regression
It used to work, and now it doesn't
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
馃悰 Describe the bug
After #80760 added MPS version of multinomial op, operations with replacement fails for arrays of more than 32K elements with non-recoverable error:
Versions
Nightly
cc @kulinseth @albanD @DenisVieriu97 @razarmehr @abhudev
The text was updated successfully, but these errors were encountered: