pytorch
1a0738f5 - [MPS] Add support for torch.linalg.cross (#91642)

Commit
1 year ago
[MPS] Add support for torch.linalg.cross (#91642) * Add support for torch.linalg.cross * Make use of `metal::cross` for float and half. For the other dtypes implement cross manually Pull Request resolved: https://github.com/pytorch/pytorch/pull/91642 Approved by: https://github.com/razarmehr, https://github.com/malfet
Author
Committer
Parents
Loading