From 22a4fb7f27149d59a6236738bebda9067d79bd51 Mon Sep 17 00:00:00 2001
From: Simon Perretta <simon.perretta@imgtec.com>
Date: Thu, 8 May 2025 21:42:49 +0100
Subject: [PATCH] pco: add ops needed to support fquantize2f16

Signed-off-by: Simon Perretta <simon.perretta@imgtec.com>
Acked-by: Erik Faye-Lund <erik.faye-lund@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/36412>
---
 src/imagination/pco/pco_trans_nir.c | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/src/imagination/pco/pco_trans_nir.c b/src/imagination/pco/pco_trans_nir.c
index a70defda5d7..df069f73467 100644
--- a/src/imagination/pco/pco_trans_nir.c
+++ b/src/imagination/pco/pco_trans_nir.c
@@ -2569,6 +2569,26 @@ static pco_instr *trans_alu(trans_ctx *tctx, nir_alu_instr *alu)
                       .roundzero = true);
       break;
 
+   case nir_op_f2f16_rtne:
+      assert(pco_ref_get_bits(src[0]) == 32);
+
+      instr = pco_pck(&tctx->b,
+                      pco_ref_bits(dest, 32),
+                      src[0],
+                      .rpt = 1,
+                      .pck_fmt = PCO_PCK_FMT_F16F16);
+      break;
+
+   case nir_op_f2f32:
+      assert(pco_ref_get_bits(src[0]) == 16);
+
+      instr = pco_unpck(&tctx->b,
+                        dest,
+                        pco_ref_elem(pco_ref_bits(src[0], 32), 0),
+                        .rpt = 1,
+                        .pck_fmt = PCO_PCK_FMT_F16F16);
+      break;
+
    case nir_op_f2i32_rtne:
       instr = pco_pck(&tctx->b, dest, src[0], .pck_fmt = PCO_PCK_FMT_S32);
       break;