pytorch
006e6f1a - Fix CPU vectorized eq and ne operations for complex types (#97374)

Commit
1 year ago
Fix CPU vectorized eq and ne operations for complex types (#97374) The comparison of both the real and imag parts need to be combined and then the result must have the real number be 1 for true and 0 for false while the imag number must always be 0. Fixes https://github.com/pytorch/pytorch/issues/75950 Pull Request resolved: https://github.com/pytorch/pytorch/pull/97374 Approved by: https://github.com/jgong5, https://github.com/kit1980
Author
Committer
Parents
Loading