pytorch
8dbe63c9 - [MPS] Casting int64 to int32 for reduction ops and raise warning. (#94484)

Commit
1 year ago
[MPS] Casting int64 to int32 for reduction ops and raise warning. (#94484) Currently casting it as a workaround till we have full support in OS. Fixes #https://github.com/pytorch/pytorch/pull/88319#issuecomment-1424010624 Pull Request resolved: https://github.com/pytorch/pytorch/pull/94484 Approved by: https://github.com/razarmehr
Author
Committer
Parents
Loading