commit 61721fd94d3eca8de3268b4c48a9a1e2a5137048
author Johannes Reifferscheid <jreiffers@google.com> Mon Apr 15 12:04:36 2024 +0200
committer Copybara-Service <copybara-worker@google.com> Mon Apr 15 03:08:06 2024 -0700
tree eb557f84e9fc330bc33751f9985ef66bf3ee492a
parent 5c289794fedc447518e163949ac218140d6411ac

Fix rsqrt inaccuracies. (#88707)

The current lowering has issues with large/subnormal values. This po
XLA's lowering and was verified using XLA's test suite and the
MLIR-based emitters.

This updates https://github.com/llvm/llvm-project/pull/88691 to also
update the correctness test for rsqrt(0). I checked C++ and Python, they
both agree the result should be (inf, nan). Updated the correctness test
to match this.

GitOrigin-RevId: 33e60f358cc0c564e771c0234dd8a60f3bddce5d