pytorch
072311bb - Enable torch._prims.amax/amin for nvFuser executor (#80070)

Commit
3 years ago
Enable torch._prims.amax/amin for nvFuser executor (#80070) This PR adds nvFuser implementations for `torch._prims.amax` and `torch._prims.amin` reduction functions. Currently, nvFuser refuses to reduce the 0d tensor, so these inputs are skipped in tests for now. An accompanying fix replaces `collections.Sequence` -> `collections.abc.Sequence` in refs because `collections.Sequence` is deprecated and removed in Python 3.10 Many ops that were skipped for the nvFuser executor test are now enabled. Pull Request resolved: https://github.com/pytorch/pytorch/pull/80070 Approved by: https://github.com/ngimel
Author
Committer
Parents
Loading