asahi: Handle load_sampler_handle

Signed-off-by: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/26056>

src/gallium/drivers/asahi/agx_nir_lower_sysvals.c

@@ -127,6 +127,9 @@ lower_intrinsic(nir_builder *b, nir_intrinsic_instr *intr)
       return load_ubo(b, intr, s->ubo_base);
    case nir_intrinsic_load_texture_handle_agx:
       return load_texture_handle(b, intr, &s->texture_base);
+   case nir_intrinsic_load_sampler_handle_agx:
+      return load_sysval_indirect(b, 1, 16, stage_table(b), &s->sampler_handle,
+                                  intr->src[0].ssa);
    case nir_intrinsic_load_vbo_base_agx:
       return load_sysval_indirect(b, 1, 64, AGX_SYSVAL_TABLE_ROOT, &u->vbo_base,
                                   intr->src[0].ssa);

src/gallium/drivers/asahi/agx_state.h

@@ -157,6 +157,9 @@ struct PACKED agx_stage_uniforms {
    uint64_t ssbo_base[PIPE_MAX_SHADER_BUFFERS];
    uint32_t ssbo_size[PIPE_MAX_SHADER_BUFFERS];
 
+   /* If lowered to bindless, sampler index in the heap */
+   uint16_t sampler_handle[PIPE_MAX_SAMPLERS];
+
    /* LOD bias as float16 */
    uint16_t lod_bias[PIPE_MAX_SAMPLERS];
 };