r600g, radeonsi: fix primitives-generated query with disabled streamout

[mesa.git] / src / gallium / drivers / radeon / r600_query.c

diff --git a/src/gallium/drivers/radeon/r600_query.c b/src/gallium/drivers/radeon/r600_query.c
index a5d75717797f6ecb3eeaaf9dd6fcf138638e69fa..e9218a2e8ecd4ea68c2f9e72c6efa641c3370ce4 100644 (file)
--- a/src/gallium/drivers/radeon/r600_query.c
+++ b/src/gallium/drivers/radeon/r600_query.c
@@ -152,6 +152,7 @@ static void r600_emit_query_begin(struct r600_common_context *ctx, struct r600_q
        uint64_t va;
 
        r600_update_occlusion_query_state(ctx, query->type, 1);
+       r600_update_prims_generated_query_state(ctx, query->type, 1);
        ctx->need_gfx_cs_space(&ctx->b, query->num_cs_dw * 2, TRUE);
 
        /* Get a new query buffer if needed. */
@@ -284,6 +285,7 @@ static void r600_emit_query_end(struct r600_common_context *ctx, struct r600_que
        }
 
        r600_update_occlusion_query_state(ctx, query->type, -1);
+       r600_update_prims_generated_query_state(ctx, query->type, -1);
 }
 
 static void r600_emit_query_predication(struct r600_common_context *ctx, struct r600_query *query,