radv: fix a memleak with GS copy shader NIR

Really need to refactor this code...

Cc: mesa-stable
Signed-off-by: Samuel Pitoiset <samuel.pitoiset@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/36317>
This commit is contained in:
Samuel Pitoiset 2025-07-23 15:46:55 +02:00 committed by Marge Bot
parent 8308e394ef
commit 0cd745c386
2 changed files with 14 additions and 0 deletions

View file

@ -2848,6 +2848,10 @@ radv_graphics_pipeline_state_finish(struct radv_device *device, struct radv_grap
if (gfx_state->stages) {
for (uint32_t i = 0; i < MESA_VULKAN_SHADER_STAGES; i++)
ralloc_free(gfx_state->stages[i].nir);
if (gfx_state->stages[MESA_SHADER_GEOMETRY].gs_copy_shader)
ralloc_free(gfx_state->stages[MESA_SHADER_GEOMETRY].gs_copy_shader);
free(gfx_state->stages);
}
}
@ -2914,6 +2918,7 @@ radv_generate_graphics_pipeline_state(struct radv_device *device, const VkGraphi
for (unsigned i = 0; i < MESA_VULKAN_SHADER_STAGES; i++) {
gfx_state->stages[i].stage = MESA_SHADER_NONE;
gfx_state->stages[i].nir = NULL;
gfx_state->stages[i].gs_copy_shader = NULL;
gfx_state->stages[i].spirv.size = 0;
gfx_state->stages[i].next_stage = MESA_SHADER_NONE;
}

View file

@ -128,6 +128,7 @@ radv_shader_object_init_graphics(struct radv_shader_object *shader_obj, struct r
for (unsigned i = 0; i < MESA_VULKAN_SHADER_STAGES; i++) {
stages[i].stage = MESA_SHADER_NONE;
stages[i].gs_copy_shader = NULL;
stages[i].nir = NULL;
stages[i].spirv.size = 0;
stages[i].next_stage = MESA_SHADER_NONE;
@ -162,6 +163,8 @@ radv_shader_object_init_graphics(struct radv_shader_object *shader_obj, struct r
binary = binaries[stage];
ralloc_free(stages[stage].nir);
if (stages[MESA_SHADER_GEOMETRY].gs_copy_shader)
ralloc_free(stages[MESA_SHADER_GEOMETRY].gs_copy_shader);
shader_obj->shader = shader;
shader_obj->binary = binary;
@ -189,6 +192,8 @@ radv_shader_object_init_graphics(struct radv_shader_object *shader_obj, struct r
binary = binaries[stage];
ralloc_free(stages[stage].nir);
if (stages[MESA_SHADER_GEOMETRY].gs_copy_shader)
ralloc_free(stages[MESA_SHADER_GEOMETRY].gs_copy_shader);
if (stage == MESA_SHADER_VERTEX) {
if (next_stage == MESA_SHADER_TESS_CTRL) {
@ -414,6 +419,7 @@ radv_shader_object_create_linked(VkDevice _device, uint32_t createInfoCount, con
for (unsigned i = 0; i < MESA_VULKAN_SHADER_STAGES; i++) {
stages[i].stage = MESA_SHADER_NONE;
stages[i].gs_copy_shader = NULL;
stages[i].nir = NULL;
stages[i].spirv.size = 0;
stages[i].next_stage = MESA_SHADER_NONE;
@ -539,6 +545,9 @@ radv_shader_object_create_linked(VkDevice _device, uint32_t createInfoCount, con
pShaders[i] = radv_shader_object_to_handle(shader_obj);
}
if (stages[MESA_SHADER_GEOMETRY].gs_copy_shader)
ralloc_free(stages[MESA_SHADER_GEOMETRY].gs_copy_shader);
return VK_SUCCESS;
}