OpenHarmony/third_party_mesa3d
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

nir: fuse ffma for atan range fixup

Browse Source 
Signed-off-by: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Reviewed-by: Ian Romanick <ian.d.romanick@intel.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/30934>
This commit is contained in:
Alyssa Rosenzweig
2024-08-29 19:25:26 -04:00
committed by Marge Bot
parent 47e7cd268c
commit 398e1ad46c
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/compiler/nir/nir_builtin_builder.c
View File

@@ -199,11 +199,13 @@ nir_atan(nir_builder *b, nir_def *y_over_x)
res = nir_ffma_imm2(b, res, x_2, coeffs[i]);
}
nir_def *tmp = nir_fmul(b, nir_fabs(b, u), res);
/* range-reduction fixup value */
nir_def *bias = nir_bcsel(b, nir_flt(b, one, abs_y_over_x),
nir_imm_floatN_t(b, -M_PI_2, bit_size),
nir_imm_floatN_t(b, 0, bit_size));
/* range-reduction fixup */
tmp = nir_bcsel(b, nir_flt(b, one, abs_y_over_x),
nir_fadd_imm(b, tmp, -M_PI_2), tmp);
/* multiply through by x while fixing up the range reduction */
nir_def *tmp = nir_ffma(b, nir_fabs(b, u), res, bias);
/* sign fixup */
nir_def *result = nir_copysign(b, tmp, y_over_x);