From 58cc66238a15184099aaa15a638d2508ba1e2bba Mon Sep 17 00:00:00 2001
From: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Date: Mon, 14 Jul 2025 19:10:21 -0400
Subject: [PATCH] nir/opt_preamble: add sampler class

AGX will use.

Signed-off-by: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Reviewed-by: Karol Herbst <kherbst@redhat.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/36127>
---
 src/compiler/nir/nir.h       | 1 +
 src/compiler/nir/nir_print.c | 9 +++++++--
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/compiler/nir/nir.h b/src/compiler/nir/nir.h
index b9a68969163..4376a1ca812 100644
--- a/src/compiler/nir/nir.h
+++ b/src/compiler/nir/nir.h
@@ -231,6 +231,7 @@ typedef enum {
 typedef enum {
    nir_preamble_class_general,
    nir_preamble_class_image,
+   nir_preamble_class_sampler,
    nir_preamble_num_classes,
 } nir_preamble_class;
 
diff --git a/src/compiler/nir/nir_print.c b/src/compiler/nir/nir_print.c
index 15eb7094354..b2ac889a1a4 100644
--- a/src/compiler/nir/nir_print.c
+++ b/src/compiler/nir/nir_print.c
@@ -1709,8 +1709,13 @@ print_intrinsic_instr(nir_intrinsic_instr *instr, print_state *state)
       case NIR_INTRINSIC_PREAMBLE_CLASS: {
          /* "General" handled above */
          nir_preamble_class cls = nir_intrinsic_preamble_class(instr);
-         assert(cls == nir_preamble_class_image);
-         fprintf(fp, "class=image");
+         if (cls == nir_preamble_class_image)
+            fprintf(fp, "class=image");
+         else if (cls == nir_preamble_class_sampler)
+            fprintf(fp, "class=sampler");
+         else
+            unreachable("invalid class");
+
          break;
       }