pytorch
49189401 - [inductor] Fix nan-handling of max and min reductions (#100572)

Commit
1 year ago
[inductor] Fix nan-handling of max and min reductions (#100572) This adds helpers that replace tritons `minimum`, `maximum`, `min` and `max` with the correct NaN prrpagation. I also removed `ops.int_minimum` in favor of `ops.minimum` because we can just omit the nan-checks by checking the dtype. Pull Request resolved: https://github.com/pytorch/pytorch/pull/100572 Approved by: https://github.com/ngimel
Author
Committer
Parents
Loading