mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Follow up to https://github.com/pytorch/pytorch/pull/162620. Add half support, as well. This fixes some failures in inductor benchmarks such as from this log https://github.com/pytorch/pytorch/actions/runs/18051942373/job/51376749459. `NotImplementedError: "aminmax_kernel" not implemented for 'Half'` Pull Request resolved: https://github.com/pytorch/pytorch/pull/164175 Approved by: https://github.com/malfet, https://github.com/jerryzh168