X-Git-Url: https://git.libre-soc.org/?p=mesa.git;a=blobdiff_plain;f=src%2Fgallium%2Fdrivers%2Fradeonsi%2Fsi_compute_prim_discard.c;h=c8c8f1159db9b08d0e3ec9a164a5a2fb5c4a963f;hp=757fc8b031b7ee69bcd3c13cf57dcd45835d87a4;hb=b4b323c81ee8ea6e756f4c5110f7ac77545325f1;hpb=b1cb72c44909fe971304da3f06e3910dea754e65

diff --git a/src/gallium/drivers/radeonsi/si_compute_prim_discard.c b/src/gallium/drivers/radeonsi/si_compute_prim_discard.c
index 757fc8b031b..c8c8f1159db 100644
--- a/src/gallium/drivers/radeonsi/si_compute_prim_discard.c
+++ b/src/gallium/drivers/radeonsi/si_compute_prim_discard.c
@@ -347,9 +347,12 @@ void si_build_prim_discard_compute_shader(struct si_shader_context *ctx)
 
    /* Create the compute shader function. */
    unsigned old_type = ctx->type;
+   gl_shader_stage old_stage = ctx->stage;
    ctx->type = PIPE_SHADER_COMPUTE;
+   ctx->stage = MESA_SHADER_COMPUTE;
    si_llvm_create_func(ctx, "prim_discard_cs", NULL, 0, THREADGROUP_SIZE);
    ctx->type = old_type;
+   ctx->stage = old_stage;
 
    if (VERTEX_COUNTER_GDS_MODE == 2) {
       ac_llvm_add_target_dep_function_attr(ctx->main_fn, "amdgpu-gds-size", 256);