From 6b52b371df9a9921728d0e058ab5979e1507f2ff Mon Sep 17 00:00:00 2001
From: Qiang Yu <yuq825@gmail.com>
Date: Fri, 20 Jun 2025 17:30:42 +0800
Subject: [PATCH] gallium/trace: dump mesh shader queries

Reviewed-by: Mike Blumenkrantz <michael.blumenkrantz@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/36976>
---
 src/gallium/auxiliary/driver_trace/tr_dump_state.c | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/gallium/auxiliary/driver_trace/tr_dump_state.c b/src/gallium/auxiliary/driver_trace/tr_dump_state.c
index 2edf016f7d4..0ba8ebfc322 100644
--- a/src/gallium/auxiliary/driver_trace/tr_dump_state.c
+++ b/src/gallium/auxiliary/driver_trace/tr_dump_state.c
@@ -1100,6 +1100,9 @@ trace_dump_query_result(unsigned query_type, unsigned index,
       trace_dump_member(uint, &result->pipeline_statistics, hs_invocations);
       trace_dump_member(uint, &result->pipeline_statistics, ds_invocations);
       trace_dump_member(uint, &result->pipeline_statistics, cs_invocations);
+      trace_dump_member(uint, &result->pipeline_statistics, ts_invocations);
+      trace_dump_member(uint, &result->pipeline_statistics, ms_invocations);
+      trace_dump_member(uint, &result->pipeline_statistics, ms_primitives);
       trace_dump_struct_end();
       break;
 
@@ -1139,6 +1142,15 @@ trace_dump_query_result(unsigned query_type, unsigned index,
       case PIPE_STAT_QUERY_CS_INVOCATIONS:
          trace_dump_member(uint, &result->pipeline_statistics, cs_invocations);
          break;
+      case PIPE_STAT_QUERY_TS_INVOCATIONS:
+         trace_dump_member(uint, &result->pipeline_statistics, ts_invocations);
+         break;
+      case PIPE_STAT_QUERY_MS_INVOCATIONS:
+         trace_dump_member(uint, &result->pipeline_statistics, ms_invocations);
+         break;
+      case PIPE_STAT_QUERY_MS_PRIMITIVES:
+         trace_dump_member(uint, &result->pipeline_statistics, ms_primitives);
+         break;
       }
       trace_dump_struct_end();
       break;