Skip to content

[MPS] Add support for aten::nansum for MPS backend #86809

@DenisVieriu97

Description

@DenisVieriu97

🐛 Describe the bug

First time contributors are welcome! 🙂
Add support for aten::nansum for MPS backend. Generic support for adding operations to MPS backend is captured here: https://github.com/pytorch/pytorch/wiki/MPS-Backend#adding-op-for-mps-backend

Versions

N/A

cc @kulinseth @albanD @malfet @razarmehr @abhudev

Metadata

Metadata

Assignees

Labels

good first issuemodule: mpsRelated to Apple Metal Performance Shaders frameworktriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions