fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2026-01-03 02:50:16 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

nir: Add NIR_OP_IS_DERIVATIVE property

Browse source 
Like IS_SELECTION.

Signed-off-by: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Reviewed-by: Daniel Schürmann <daniel@schuermann.dev>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/24833>
This commit is contained in:
Alyssa Rosenzweig 2023-08-24 07:04:20 -04:00
parent f6f9c07efe
commit b77dc9f7d7
2 changed files with 13 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/compiler/nir/nir.h
View file

@ -1358,6 +1358,12 @@ typedef enum {
* comparison.
*/
NIR_OP_IS_SELECTION = (1 << 2),
/**
* Operation where a screen-space derivative is taken of src[0]. Must not be
* moved into non-uniform control flow.
*/
NIR_OP_IS_DERIVATIVE = (1 << 3),
} nir_op_algebraic_property;
/* vec16 is the widest ALU op in NIR, making the max number of input of ALU

11
src/compiler/nir/nir_opcodes.py
View file

@ -150,6 +150,7 @@ def type_base_type(type_):
_2src_commutative = "2src_commutative "
associative = "associative "
selection = "selection "
derivative = "derivative "
# global dictionary of opcodes
opcodes = {}
@ -164,8 +165,9 @@ def opcode(name, output_size, output_type, input_sizes, input_types,
def unop_convert(name, out_type, in_type, const_expr, description = ""):
opcode(name, 0, out_type, [0], [in_type], False, "", const_expr, description)
def unop(name, ty, const_expr, description = ""):
opcode(name, 0, ty, [0], [ty], False, "", const_expr, description)
def unop(name, ty, const_expr, description = "", algebraic_properties = ""):
opcode(name, 0, ty, [0], [ty], False, algebraic_properties, const_expr,
description)
def unop_horiz(name, output_size, output_type, input_size, input_type,
const_expr, description = ""):
@ -338,6 +340,7 @@ of a constant is 0 if the constant is not Inf or NaN.
for mode, suffix in [("either fine or coarse", ""), ("fine", "_fine"), ("coarse", "_coarse")]:
for axis in ["x", "y"]:
unop(f"fdd{axis}{suffix}", tfloat, "isfinite(src0) ? 0.0 : NAN",
algebraic_properties = derivative,
description = deriv_template.format(mode, axis.upper()))
# Floating point pack and unpack operations.
@ -1406,8 +1409,8 @@ opcode("b32fcsel_mdg", 0, tuint, [0, 0, 0],
""")
# Magnitude equal to fddx/y, sign undefined. Derivative of a constant is zero.
unop("fddx_must_abs_mali", tfloat, "0.0")
unop("fddy_must_abs_mali", tfloat, "0.0")
unop("fddx_must_abs_mali", tfloat, "0.0", algebraic_properties = "derivative")
unop("fddy_must_abs_mali", tfloat, "0.0", algebraic_properties = "derivative")
# DXIL specific double [un]pack
# DXIL doesn't support generic [un]pack instructions, so we want those