pytorch
ee934fd6 - Use unordered NEQ comparison for vec512 operator!= implementations (#97466)

Commit
1 year ago
Use unordered NEQ comparison for vec512 operator!= implementations (#97466) This is consistent with the vec256 operator!= implementations. _CMP_NEQ_UQ is the logical opposite of _CMP_EQ_OQ comparison used in the operator== implementations. Using the ordered NEQ operation results in nan != nan being false which is incorrect. Pull Request resolved: https://github.com/pytorch/pytorch/pull/97466 Approved by: https://github.com/jgong5, https://github.com/sanchitintel
Author
Committer
Parents
Loading