From 99a636e434230bb4c40f28915149369f98c272ff Mon Sep 17 00:00:00 2001 From: Lorenzo Rossi Date: Wed, 6 May 2026 18:18:51 +0200 Subject: [PATCH] nir/opt_algebraic: Optimize mediump ops done in highp When we do f2fmp(fadd(f2f32(a), f2f32(b))) we can always optimize it to fadd(a, b) and obtain the same bit-precise result, I verified this using a custom script with Berkley SoftFloat. This commit enables these optimizations for fadd, fmul, fdiv and frem, which have all been proven to be bit-by-bit identical. --- src/compiler/nir/nir_opt_algebraic.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/src/compiler/nir/nir_opt_algebraic.py b/src/compiler/nir/nir_opt_algebraic.py index 9d0cd043504..1e4d4ed4019 100644 --- a/src/compiler/nir/nir_opt_algebraic.py +++ b/src/compiler/nir/nir_opt_algebraic.py @@ -1963,11 +1963,6 @@ optimizations.extend([ (('f2u32', ('f2fmp', 'a@32')), ('f2u32', a), 'true', TestStatus.UNSUPPORTED), (('i2f32', ('i2imp', 'a@32')), ('i2f32', a), 'true', TestStatus.UNSUPPORTED), - # f16 -> f2f32 -> fmul32 -> f2fmp when the second operand is a constant - # The optimization only works when the constant can be safely represented with 16 bits - (('f2fmp', ('fmul(is_used_once,contract)', ('f2f32', 'a@16'), '#b(is_representable_as_f16)')), ('fmul', a, ('f2fmp', b)), 'true', TestStatus.UNSUPPORTED), - (('f2fmp', ('fadd(is_used_once,contract)', ('f2f32', 'a@16'), '#b(is_representable_as_f16)')), ('fadd', a, ('f2fmp', b)), 'true', TestStatus.UNSUPPORTED), - (('ffloor', 'a(is_integral)'), a), (('fceil', 'a(is_integral)'), a), (('ftrunc', 'a(is_integral)'), a), @@ -2011,6 +2006,14 @@ optimizations.extend([ (('ult', a, 0), False), ]) +# f2fmp(op(f2f16(a), f2f16(b))) -> op(a, b) +# The result is bit-per-bit identical except for NaN representations. +for op in ['fadd', 'fmul', 'fdiv', 'frem']: + optimizations += [ + (('f2fmp', (f'{op}(is_used_once)', ('f2f32', 'a@16'), '#b(is_representable_as_f16)')), (op, a, ('f2fmp', b)), 'true', TestStatus.UNSUPPORTED), + (('f2fmp', (f'{op}(is_used_once)', ('f2f32', 'a@16'), ('f2f32', 'b@16'))), (op, a, b), 'true', TestStatus.UNSUPPORTED), + ] + for bits in [16, 32, 64]: cond = '!(options->lower_doubles_options & nir_lower_fp64_full_software)' if bits == 64 else 'true' bcsel = 'bcsel@{}'.format(bits)