From c20d5ee3c236d81fb9f09ed7a7421bb912019392 Mon Sep 17 00:00:00 2001
From: Samuel Pitoiset <samuel.pitoiset@gmail.com>
Date: Thu, 15 Sep 2022 14:48:17 +0200
Subject: [PATCH] radv: lower nir_load_rasterization_samples_amd in ABI

Signed-off-by: Samuel Pitoiset <samuel.pitoiset@gmail.com>
Reviewed-by: Rhys Perry <pendingchaos02@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/18677>
---
 src/amd/vulkan/radv_nir_lower_abi.c | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/amd/vulkan/radv_nir_lower_abi.c b/src/amd/vulkan/radv_nir_lower_abi.c
index ed77952da3d..eade6b4d18e 100644
--- a/src/amd/vulkan/radv_nir_lower_abi.c
+++ b/src/amd/vulkan/radv_nir_lower_abi.c
@@ -275,6 +275,13 @@ lower_abi_instr(nir_builder *b, nir_instr *instr, void *state)
                                         .base = sample_pos_offset, .access = ACCESS_NON_WRITEABLE);
       break;
    }
+   case nir_intrinsic_load_rasterization_samples_amd:
+      if (s->pl_key->dynamic_rasterization_samples) {
+         replacement = ac_nir_load_arg(b, &s->args->ac, s->args->ps_num_samples);
+      } else {
+         replacement = nir_imm_int(b, s->pl_key->ps.num_samples);
+      }
+      break;
    default:
       break;
    }