pytorch
19264b50 - [MPS] Add support for nansum on mps (#93845)

Commit
1 year ago
[MPS] Add support for nansum on mps (#93845) * Add `nansum_out_mps` and `nansum_mps` functions * Moved `get_dtype_from_self` into ReduceOpsUtils.h Fixes #86809 Pull Request resolved: https://github.com/pytorch/pytorch/pull/93845 Approved by: https://github.com/malfet
Author
Committer
Parents
Loading