From 22b37c16c8a627ea8120e8a5cddbc44288c44869 Mon Sep 17 00:00:00 2001
From: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Date: Tue, 15 Jul 2025 15:33:24 -0400
Subject: [PATCH] nir: add nir_alu_src_rewrite_scalar helper

this is a little annoying.

Signed-off-by: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Reviewed-by: Mel Henning <mhenning@darkrefraction.com>
Acked-by: Adam Jackson <ajax@redhat.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/36147>
---
 src/compiler/nir/nir.h | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/compiler/nir/nir.h b/src/compiler/nir/nir.h
index 5771930c7e4..7cc6d4b8caa 100644
--- a/src/compiler/nir/nir.h
+++ b/src/compiler/nir/nir.h
@@ -4378,6 +4378,13 @@ nir_src_rewrite(nir_src *src, nir_def *new_ssa)
    list_addtail(&src->use_link, &new_ssa->uses);
 }
 
+static inline void
+nir_alu_src_rewrite_scalar(nir_alu_src *alu, nir_scalar s)
+{
+   nir_src_rewrite(&alu->src, s.def);
+   alu->swizzle[0] = (uint8_t)s.comp;
+}
+
 /** Initialize a nir_src
  *
  * This is almost never the helper you want to use.  This helper assumes that