From fd9c2ce73d05eaed611519b997aa45f7622b691e Mon Sep 17 00:00:00 2001
From: Faith Ekstrand <faith.ekstrand@collabora.com>
Date: Wed, 27 May 2026 23:19:04 -0400
Subject: [PATCH] compiler/rust/nir: Add helpers for getting ALU input/output
 types

Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/41896>
---
 src/compiler/rust/nir.rs | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/compiler/rust/nir.rs b/src/compiler/rust/nir.rs
index 1ffc4b8d385..cb50049f229 100644
--- a/src/compiler/rust/nir.rs
+++ b/src/compiler/rust/nir.rs
@@ -224,6 +224,14 @@ impl nir_alu_instr {
     pub fn get_src(&self, idx: usize) -> &nir_alu_src {
         &self.srcs_as_slice()[idx]
     }
+
+    pub fn output_type(&self) -> ALUType {
+        ALUType(self.info().output_type)
+    }
+
+    pub fn input_type(&self, src_idx: usize) -> ALUType {
+        ALUType(self.info().input_types[src_idx])
+    }
 }
 
 impl nir_op_info {