From bf6c3e9d99cfa7b6567b0f7312a07a913bba350d Mon Sep 17 00:00:00 2001
From: Karol Herbst <kherbst@redhat.com>
Date: Tue, 24 Feb 2026 04:10:17 +0100
Subject: [PATCH] nak: add is_gpr_reg and is_ugpr_reg helpers

Reviewed-by: Mel Henning <mhenning@darkrefraction.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/40891>
---
 src/nouveau/compiler/nak/ir.rs | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/src/nouveau/compiler/nak/ir.rs b/src/nouveau/compiler/nak/ir.rs
index 4065c72c667..8d5feffb9c7 100644
--- a/src/nouveau/compiler/nak/ir.rs
+++ b/src/nouveau/compiler/nak/ir.rs
@@ -1158,6 +1158,24 @@ impl Src {
         }
     }
 
+    #[allow(dead_code)]
+    pub fn is_gpr_reg(&self) -> bool {
+        match &self.src_ref {
+            SrcRef::SSA(ssa) => ssa.file() == RegFile::GPR,
+            SrcRef::Reg(reg) => reg.file() == RegFile::GPR,
+            _ => false,
+        }
+    }
+
+    #[allow(dead_code)]
+    pub fn is_ugpr_reg(&self) -> bool {
+        match &self.src_ref {
+            SrcRef::SSA(ssa) => ssa.file() == RegFile::UGPR,
+            SrcRef::Reg(reg) => reg.file() == RegFile::UGPR,
+            _ => false,
+        }
+    }
+
     pub fn is_predicate(&self) -> bool {
         self.src_ref.is_predicate()
     }