fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2025-12-28 06:00:10 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

nir/opt_constant_folding: handle derivative intrinsics

Browse source 
Signed-off-by: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Reviewed-by: Jesse Natalie <jenatali@microsoft.com>
Reviewed-by: Georg Lehmann <dadschoorse@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/30565>
This commit is contained in:
Alyssa Rosenzweig 2024-07-23 17:01:57 -04:00 • committed by Marge Bot
parent e0cc041674
commit 66724e28ac
1 changed files with 25 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/compiler/nir/nir_opt_constant_folding.c
View file

@ -247,6 +247,31 @@ try_fold_intrinsic(nir_builder *b, nir_intrinsic_instr *intrin,
return true;
}
case nir_intrinsic_ddx:
case nir_intrinsic_ddx_fine:
case nir_intrinsic_ddx_coarse:
case nir_intrinsic_ddy:
case nir_intrinsic_ddy_fine:
case nir_intrinsic_ddy_coarse: {
if (!nir_src_is_const(intrin->src[0]))
return false;
/* Derivative of a constant is zero, except for NaNs and Infs */
nir_const_value imm[NIR_MAX_VEC_COMPONENTS];
unsigned sz = intrin->def.bit_size;
b->cursor = nir_before_instr(&intrin->instr);
for (unsigned i = 0; i < intrin->def.num_components; i++) {
bool finite = isfinite(nir_src_comp_as_float(intrin->src[0], i));
imm[i] = nir_const_value_for_float(finite ? 0 : NAN, sz);
}
nir_def_replace(&intrin->def,
nir_build_imm(b, intrin->def.num_components, sz, imm));
return true;
}
case nir_intrinsic_vote_any:
case nir_intrinsic_vote_all:
case nir_intrinsic_read_invocation: