radeonsi: lower task/mesh shader io to mem

Reviewed-by: Marek Olšák <marek.olsak@amd.com> Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/37505>
2025-04-28 14:52:04 +08:00
parent 5931dbf7ac
commit 4c315bdbfa
1 changed files with 8 additions and 0 deletions
@@ -335,6 +335,14 @@ static void si_lower_nir(struct si_screen *sscreen, struct nir_shader *nir)
      }

      NIR_PASS(_, nir, nir_lower_gs_intrinsics, flags);
+   } else if (nir->info.stage == MESA_SHADER_TASK) {
+      NIR_PASS(_, nir, ac_nir_lower_task_outputs_to_mem,
+               sscreen->task_info.payload_entry_size,
+               sscreen->task_info.num_entries, false);
+   } else if (nir->info.stage == MESA_SHADER_MESH) {
+      NIR_PASS(_, nir, ac_nir_lower_mesh_inputs_to_mem,
+               sscreen->task_info.payload_entry_size,
+               sscreen->task_info.num_entries);
   }

   if (mesa_shader_stage_is_compute(nir->info.stage)) {