Mesa (master): radeonsi: implement SQTT support

GitLab Mirror gitlab-mirror at kemper.freedesktop.org
Thu Jan 7 09:30:10 UTC 2021


Module: Mesa
Branch: master
Commit: 07c1504d1b08d4f61aeada7364d63e26d05d05f9
URL:    http://cgit.freedesktop.org/mesa/mesa/commit/?id=07c1504d1b08d4f61aeada7364d63e26d05d05f9

Author: Pierre-Eric Pelloux-Prayer <pierre-eric.pelloux-prayer at amd.com>
Date:   Tue Dec  8 13:08:10 2020 +0100

radeonsi: implement SQTT support

Reviewed-by: Marek Olšák <marek.olsak at amd.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/8002>

---

 src/gallium/drivers/radeonsi/Makefile.sources  |   1 +
 src/gallium/drivers/radeonsi/meson.build       |   1 +
 src/gallium/drivers/radeonsi/si_compute.c      |   5 +
 src/gallium/drivers/radeonsi/si_gfx_cs.c       |   5 +
 src/gallium/drivers/radeonsi/si_pipe.c         |  11 +
 src/gallium/drivers/radeonsi/si_pipe.h         |  17 +
 src/gallium/drivers/radeonsi/si_sqtt.c         | 707 +++++++++++++++++++++++++
 src/gallium/drivers/radeonsi/si_state_draw.cpp |  19 +
 8 files changed, 766 insertions(+)

diff --git a/src/gallium/drivers/radeonsi/Makefile.sources b/src/gallium/drivers/radeonsi/Makefile.sources
index d9235192190..9a3736c6c23 100644
--- a/src/gallium/drivers/radeonsi/Makefile.sources
+++ b/src/gallium/drivers/radeonsi/Makefile.sources
@@ -37,6 +37,7 @@ C_SOURCES := \
 	si_shader_llvm_vs.c \
 	si_shader_nir.c \
 	si_shaderlib_tgsi.c \
+	si_sqtt.c \
 	si_state.c \
 	si_state_binning.c \
 	si_state_draw.cpp \
diff --git a/src/gallium/drivers/radeonsi/meson.build b/src/gallium/drivers/radeonsi/meson.build
index 520aa293be7..8d7538dec28 100644
--- a/src/gallium/drivers/radeonsi/meson.build
+++ b/src/gallium/drivers/radeonsi/meson.build
@@ -57,6 +57,7 @@ files_libradeonsi = files(
   'si_shader_llvm_vs.c',
   'si_shader_nir.c',
   'si_shaderlib_tgsi.c',
+  'si_sqtt.c',
   'si_state.c',
   'si_state.h',
   'si_state_binning.c',
diff --git a/src/gallium/drivers/radeonsi/si_compute.c b/src/gallium/drivers/radeonsi/si_compute.c
index a0213015406..4289e83083b 100644
--- a/src/gallium/drivers/radeonsi/si_compute.c
+++ b/src/gallium/drivers/radeonsi/si_compute.c
@@ -793,6 +793,11 @@ static void si_emit_dispatch_packets(struct si_context *sctx, const struct pipe_
       radeon_emit(cs, info->grid[2]);
       radeon_emit(cs, dispatch_initiator);
    }
+
+   if (unlikely(sctx->thread_trace_enabled && sctx->chip_class >= GFX9)) {
+      radeon_emit(&sctx->gfx_cs, PKT3(PKT3_EVENT_WRITE, 0, 0));
+      radeon_emit(&sctx->gfx_cs, EVENT_TYPE(V_028A90_THREAD_TRACE_MARKER) | EVENT_INDEX(0));
+   }
 }
 
 static void si_launch_grid(struct pipe_context *ctx, const struct pipe_grid_info *info)
diff --git a/src/gallium/drivers/radeonsi/si_gfx_cs.c b/src/gallium/drivers/radeonsi/si_gfx_cs.c
index 6ac1233e3a9..42d895de91e 100644
--- a/src/gallium/drivers/radeonsi/si_gfx_cs.c
+++ b/src/gallium/drivers/radeonsi/si_gfx_cs.c
@@ -215,6 +215,11 @@ void si_flush_gfx_cs(struct si_context *ctx, unsigned flags, struct pipe_fence_h
       si_check_vm_faults(ctx, &ctx->current_saved_cs->gfx, RING_GFX);
    }
 
+   if (unlikely(ctx->thread_trace &&
+                (flags & PIPE_FLUSH_END_OF_FRAME))) {
+      si_handle_thread_trace(ctx, &ctx->gfx_cs);
+   }
+
    if (ctx->current_saved_cs)
       si_saved_cs_reference(&ctx->current_saved_cs, NULL);
 
diff --git a/src/gallium/drivers/radeonsi/si_pipe.c b/src/gallium/drivers/radeonsi/si_pipe.c
index 5c9775ee039..fdee2c509a2 100644
--- a/src/gallium/drivers/radeonsi/si_pipe.c
+++ b/src/gallium/drivers/radeonsi/si_pipe.c
@@ -113,6 +113,7 @@ static const struct debug_named_value radeonsi_debug_options[] = {
    {"nofmask", DBG(NO_FMASK), "Disable MSAA compression"},
 
    {"tmz", DBG(TMZ), "Force allocation of scanout/depth/stencil buffer as encrypted"},
+   {"sqtt", DBG(SQTT), "Enable SQTT"},
 
    DEBUG_NAMED_VALUE_END /* must be last */
 };
@@ -186,6 +187,9 @@ static void si_destroy_context(struct pipe_context *context)
    if (sctx->chip_class >= GFX10 && sctx->has_graphics)
       gfx10_destroy_query(sctx);
 
+   if (sctx->thread_trace)
+      si_destroy_thread_trace(sctx);
+
    pipe_resource_reference(&sctx->esgs_ring, NULL);
    pipe_resource_reference(&sctx->gsvs_ring, NULL);
    pipe_resource_reference(&sctx->tess_rings, NULL);
@@ -734,6 +738,13 @@ static struct pipe_context *si_pipe_create_context(struct pipe_screen *screen, v
 
    ctx = si_create_context(screen, flags);
 
+   if (ctx && sscreen->info.chip_class >= GFX9 && sscreen->debug_flags & DBG(SQTT)) {
+      if (!si_init_thread_trace((struct si_context *)ctx)) {
+         FREE(ctx);
+         return NULL;
+      }
+   }
+
    if (!(flags & PIPE_CONTEXT_PREFER_THREADED))
       return ctx;
 
diff --git a/src/gallium/drivers/radeonsi/si_pipe.h b/src/gallium/drivers/radeonsi/si_pipe.h
index 113c2f79676..339021a792d 100644
--- a/src/gallium/drivers/radeonsi/si_pipe.h
+++ b/src/gallium/drivers/radeonsi/si_pipe.h
@@ -31,6 +31,7 @@
 #include "util/u_idalloc.h"
 #include "util/u_suballoc.h"
 #include "util/u_threaded_context.h"
+#include "ac_sqtt.h"
 
 #ifdef __cplusplus
 extern "C" {
@@ -233,6 +234,7 @@ enum
    DBG_NO_FMASK,
 
    DBG_TMZ,
+   DBG_SQTT,
 
    DBG_COUNT
 };
@@ -1297,6 +1299,11 @@ struct si_context {
    struct hash_table *dirty_implicit_resources;
 
    pipe_draw_vbo_func draw_vbo[NUM_GFX_VERSIONS - GFX6][2][2][2][2];
+
+   /* SQTT */
+   struct ac_thread_trace_data *thread_trace;
+   struct pipe_fence_handle *last_sqtt_fence;
+   bool thread_trace_enabled;
 };
 
 /* si_blit.c */
@@ -1558,6 +1565,16 @@ bool si_texture_disable_dcc(struct si_context *sctx, struct si_texture *tex);
 void si_init_screen_texture_functions(struct si_screen *sscreen);
 void si_init_context_texture_functions(struct si_context *sctx);
 
+/* si_sqtt.c */
+void si_sqtt_write_event_marker(struct si_context* sctx, struct radeon_cmdbuf *rcs,
+                                enum rgp_sqtt_marker_event_type api_type,
+                                uint32_t vertex_offset_user_data,
+                                uint32_t instance_offset_user_data,
+                                uint32_t draw_index_user_data);
+bool si_init_thread_trace(struct si_context *sctx);
+void si_destroy_thread_trace(struct si_context *sctx);
+void si_handle_thread_trace(struct si_context *sctx, struct radeon_cmdbuf *rcs);
+
 /*
  * common helpers
  */
diff --git a/src/gallium/drivers/radeonsi/si_sqtt.c b/src/gallium/drivers/radeonsi/si_sqtt.c
new file mode 100644
index 00000000000..9bd1483ae31
--- /dev/null
+++ b/src/gallium/drivers/radeonsi/si_sqtt.c
@@ -0,0 +1,707 @@
+/*
+ * Copyright 2020 Advanced Micro Devices, Inc.
+ * All Rights Reserved.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * on the rights to use, copy, modify, merge, publish, distribute, sub
+ * license, and/or sell copies of the Software, and to permit persons to whom
+ * the Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice (including the next
+ * paragraph) shall be included in all copies or substantial portions of the
+ * Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
+ * THE AUTHOR(S) AND/OR THEIR SUPPLIERS BE LIABLE FOR ANY CLAIM,
+ * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
+ * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
+ * USE OR OTHER DEALINGS IN THE SOFTWARE.
+ *
+ */
+
+
+#include "si_pipe.h"
+#include "si_build_pm4.h"
+
+#include "ac_rgp.h"
+#include "ac_sqtt.h"
+#include "util/u_memory.h"
+
+static void
+si_emit_spi_config_cntl(struct si_context* sctx,
+                        struct radeon_cmdbuf *cs, bool enable);
+
+static inline void
+radeon_set_privileged_config_reg(struct radeon_cmdbuf *cs,
+                                 unsigned reg,
+                                 unsigned value)
+{
+   assert(reg < CIK_UCONFIG_REG_OFFSET);
+
+   radeon_emit(cs, PKT3(PKT3_COPY_DATA, 4, 0));
+   radeon_emit(cs, COPY_DATA_SRC_SEL(COPY_DATA_IMM) |
+               COPY_DATA_DST_SEL(COPY_DATA_PERF));
+   radeon_emit(cs, value);
+   radeon_emit(cs, 0); /* unused */
+   radeon_emit(cs, reg >> 2);
+   radeon_emit(cs, 0); /* unused */
+}
+
+static bool
+si_thread_trace_init_bo(struct si_context *sctx)
+{
+   struct radeon_winsys *ws = sctx->ws;
+   uint64_t size;
+
+   /* The buffer size and address need to be aligned in HW regs. Align the
+    * size as early as possible so that we do all the allocation & addressing
+    * correctly. */
+   sctx->thread_trace->buffer_size = align64(sctx->thread_trace->buffer_size,
+                                             1u << SQTT_BUFFER_ALIGN_SHIFT);
+
+   /* Compute total size of the thread trace BO for 4 SEs. */
+   size = align64(sizeof(struct ac_thread_trace_info) * 4,
+                  1 << SQTT_BUFFER_ALIGN_SHIFT);
+   size += sctx->thread_trace->buffer_size * 4;
+
+   sctx->thread_trace->bo =
+      ws->buffer_create(ws, size, 4096,
+                        RADEON_DOMAIN_VRAM,
+                        RADEON_FLAG_NO_INTERPROCESS_SHARING |
+                        RADEON_FLAG_GTT_WC |
+                        RADEON_FLAG_NO_SUBALLOC);
+   if (!sctx->thread_trace->bo)
+      return false;
+
+   return true;
+}
+
+static void
+si_emit_thread_trace_start(struct si_context* sctx,
+                           struct radeon_cmdbuf *cs,
+                           uint32_t queue_family_index)
+{
+   struct si_screen *sscreen = sctx->screen;
+   uint32_t shifted_size = sctx->thread_trace->buffer_size >> SQTT_BUFFER_ALIGN_SHIFT;
+   unsigned max_se = sscreen->info.max_se;
+
+   for (unsigned se = 0; se < max_se; se++) {
+      uint64_t va = sctx->ws->buffer_get_virtual_address(sctx->thread_trace->bo);
+      uint64_t data_va = ac_thread_trace_get_data_va(sctx->thread_trace, va, se);
+      uint64_t shifted_va = data_va >> SQTT_BUFFER_ALIGN_SHIFT;
+
+      /* Target SEx and SH0. */
+      radeon_set_uconfig_reg(cs, R_030800_GRBM_GFX_INDEX,
+                             S_030800_SE_INDEX(se) |
+                             S_030800_SH_INDEX(0) |
+                             S_030800_INSTANCE_BROADCAST_WRITES(1));
+
+      if (sctx->chip_class == GFX10) {
+         /* Order seems important for the following 2 registers. */
+         radeon_set_privileged_config_reg(cs, R_008D04_SQ_THREAD_TRACE_BUF0_SIZE,
+                                          S_008D04_SIZE(shifted_size) |
+                                          S_008D04_BASE_HI(shifted_va >> 32));
+
+         radeon_set_privileged_config_reg(cs, R_008D00_SQ_THREAD_TRACE_BUF0_BASE,
+                                          S_008D00_BASE_LO(shifted_va));
+
+         radeon_set_privileged_config_reg(cs, R_008D14_SQ_THREAD_TRACE_MASK,
+                                          S_008D14_WTYPE_INCLUDE(0x7f) | /* all shader stages */
+                                          S_008D14_SA_SEL(0) |
+                                          S_008D14_WGP_SEL(0) |
+                                          S_008D14_SIMD_SEL(0));
+
+         radeon_set_privileged_config_reg(cs, R_008D18_SQ_THREAD_TRACE_TOKEN_MASK,
+                      S_008D18_REG_INCLUDE(V_008D18_REG_INCLUDE_SQDEC |
+                                           V_008D18_REG_INCLUDE_SHDEC |
+                                           V_008D18_REG_INCLUDE_GFXUDEC |
+                                           V_008D18_REG_INCLUDE_CONTEXT |
+                                           V_008D18_REG_INCLUDE_COMP |
+                                           V_008D18_REG_INCLUDE_CONTEXT |
+                                           V_008D18_REG_INCLUDE_CONFIG) |
+                      S_008D18_TOKEN_EXCLUDE(V_008D18_TOKEN_EXCLUDE_PERF));
+
+         /* Should be emitted last (it enables thread traces). */
+         radeon_set_privileged_config_reg(cs, R_008D1C_SQ_THREAD_TRACE_CTRL,
+                                          S_008D1C_MODE(1) |
+                                          S_008D1C_HIWATER(5) |
+                                          S_008D1C_UTIL_TIMER(1) |
+                                          S_008D1C_RT_FREQ(2) | /* 4096 clk */
+                                          S_008D1C_DRAW_EVENT_EN(1) |
+                                          S_008D1C_REG_STALL_EN(1) |
+                                          S_008D1C_SPI_STALL_EN(1) |
+                                          S_008D1C_SQ_STALL_EN(1) |
+                                          S_008D1C_REG_DROP_ON_STALL(0));
+      } else {
+         /* Order seems important for the following 4 registers. */
+         radeon_set_uconfig_reg(cs, R_030CDC_SQ_THREAD_TRACE_BASE2,
+                                S_030CDC_ADDR_HI(shifted_va >> 32));
+
+         radeon_set_uconfig_reg(cs, R_030CC0_SQ_THREAD_TRACE_BASE,
+                                S_030CC0_ADDR(shifted_va));
+
+         radeon_set_uconfig_reg(cs, R_030CC4_SQ_THREAD_TRACE_SIZE,
+                                S_030CC4_SIZE(shifted_size));
+
+         radeon_set_uconfig_reg(cs, R_030CD4_SQ_THREAD_TRACE_CTRL,
+                                S_030CD4_RESET_BUFFER(1));
+
+         uint32_t thread_trace_mask = S_030CC8_CU_SEL(2) |
+                                      S_030CC8_SH_SEL(0) |
+                                      S_030CC8_SIMD_EN(0xf) |
+                                      S_030CC8_VM_ID_MASK(0) |
+                                      S_030CC8_REG_STALL_EN(1) |
+                                      S_030CC8_SPI_STALL_EN(1) |
+                                      S_030CC8_SQ_STALL_EN(1);
+
+         radeon_set_uconfig_reg(cs, R_030CC8_SQ_THREAD_TRACE_MASK,
+                                thread_trace_mask);
+
+         /* Trace all tokens and registers. */
+         radeon_set_uconfig_reg(cs, R_030CCC_SQ_THREAD_TRACE_TOKEN_MASK,
+                                S_030CCC_TOKEN_MASK(0xbfff) |
+                                S_030CCC_REG_MASK(0xff) |
+                                S_030CCC_REG_DROP_ON_STALL(0));
+
+         /* Enable SQTT perf counters for all CUs. */
+         radeon_set_uconfig_reg(cs, R_030CD0_SQ_THREAD_TRACE_PERF_MASK,
+                                S_030CD0_SH0_MASK(0xffff) |
+                                S_030CD0_SH1_MASK(0xffff));
+
+         radeon_set_uconfig_reg(cs, R_030CE0_SQ_THREAD_TRACE_TOKEN_MASK2,
+                                S_030CE0_INST_MASK(0xffffffff));
+
+         radeon_set_uconfig_reg(cs, R_030CEC_SQ_THREAD_TRACE_HIWATER,
+                                S_030CEC_HIWATER(4));
+
+         if (sctx->chip_class == GFX9) {
+            /* Reset thread trace status errors. */
+            radeon_set_uconfig_reg(cs, R_030CE8_SQ_THREAD_TRACE_STATUS,
+                                   S_030CE8_UTC_ERROR(0));
+         }
+
+         /* Enable the thread trace mode. */
+         uint32_t thread_trace_mode =
+            S_030CD8_MASK_PS(1) |
+            S_030CD8_MASK_VS(1) |
+            S_030CD8_MASK_GS(1) |
+            S_030CD8_MASK_ES(1) |
+            S_030CD8_MASK_HS(1) |
+            S_030CD8_MASK_LS(1) |
+            S_030CD8_MASK_CS(1) |
+            S_030CD8_AUTOFLUSH_EN(1) | /* periodically flush SQTT data to memory */
+            S_030CD8_MODE(1);
+
+         if (sctx->chip_class == GFX9) {
+            /* Count SQTT traffic in TCC perf counters. */
+            thread_trace_mode |= S_030CD8_TC_PERF_EN(1);
+         }
+
+         radeon_set_uconfig_reg(cs, R_030CD8_SQ_THREAD_TRACE_MODE,
+                                thread_trace_mode);
+      }
+   }
+
+   /* Restore global broadcasting. */
+   radeon_set_uconfig_reg(cs, R_030800_GRBM_GFX_INDEX,
+                          S_030800_SE_BROADCAST_WRITES(1) |
+                             S_030800_SH_BROADCAST_WRITES(1) |
+                             S_030800_INSTANCE_BROADCAST_WRITES(1));
+
+   /* Start the thread trace with a different event based on the queue. */
+   if (queue_family_index == RING_COMPUTE) {
+      radeon_set_sh_reg(cs, R_00B878_COMPUTE_THREAD_TRACE_ENABLE,
+                        S_00B878_THREAD_TRACE_ENABLE(1));
+   } else {
+      radeon_emit(cs, PKT3(PKT3_EVENT_WRITE, 0, 0));
+      radeon_emit(cs, EVENT_TYPE(V_028A90_THREAD_TRACE_START) | EVENT_INDEX(0));
+   }
+}
+
+static const uint32_t gfx9_thread_trace_info_regs[] =
+{
+   R_030CE4_SQ_THREAD_TRACE_WPTR,
+   R_030CE8_SQ_THREAD_TRACE_STATUS,
+   R_030CF0_SQ_THREAD_TRACE_CNTR,
+};
+
+static const uint32_t gfx10_thread_trace_info_regs[] =
+{
+   R_008D10_SQ_THREAD_TRACE_WPTR,
+   R_008D20_SQ_THREAD_TRACE_STATUS,
+   R_008D24_SQ_THREAD_TRACE_DROPPED_CNTR,
+};
+
+static void
+si_copy_thread_trace_info_regs(struct si_context* sctx,
+             struct radeon_cmdbuf *cs,
+             unsigned se_index)
+{
+   const uint32_t *thread_trace_info_regs = NULL;
+
+   switch (sctx->chip_class) {
+   case GFX10:
+      thread_trace_info_regs = gfx10_thread_trace_info_regs;
+      break;
+   case GFX9:
+      thread_trace_info_regs = gfx9_thread_trace_info_regs;
+      break;
+   default:
+      unreachable("Unsupported chip_class");
+   }
+
+   /* Get the VA where the info struct is stored for this SE. */
+   uint64_t va = sctx->ws->buffer_get_virtual_address(sctx->thread_trace->bo);
+   uint64_t info_va = ac_thread_trace_get_info_va(va, se_index);
+
+   /* Copy back the info struct one DWORD at a time. */
+   for (unsigned i = 0; i < 3; i++) {
+      radeon_emit(cs, PKT3(PKT3_COPY_DATA, 4, 0));
+      radeon_emit(cs, COPY_DATA_SRC_SEL(COPY_DATA_PERF) |
+                      COPY_DATA_DST_SEL(COPY_DATA_TC_L2) |
+                  COPY_DATA_WR_CONFIRM);
+      radeon_emit(cs, thread_trace_info_regs[i] >> 2);
+      radeon_emit(cs, 0); /* unused */
+      radeon_emit(cs, (info_va + i * 4));
+      radeon_emit(cs, (info_va + i * 4) >> 32);
+   }
+}
+
+
+
+static void
+si_emit_thread_trace_stop(struct si_context *sctx,
+                          struct radeon_cmdbuf *cs,
+                          uint32_t queue_family_index)
+{
+   unsigned max_se = sctx->screen->info.max_se;
+
+   /* Stop the thread trace with a different event based on the queue. */
+   if (queue_family_index == RING_COMPUTE) {
+      radeon_set_sh_reg(cs, R_00B878_COMPUTE_THREAD_TRACE_ENABLE,
+                        S_00B878_THREAD_TRACE_ENABLE(0));
+   } else {
+      radeon_emit(cs, PKT3(PKT3_EVENT_WRITE, 0, 0));
+      radeon_emit(cs, EVENT_TYPE(V_028A90_THREAD_TRACE_STOP) | EVENT_INDEX(0));
+   }
+
+   radeon_emit(cs, PKT3(PKT3_EVENT_WRITE, 0, 0));
+   radeon_emit(cs, EVENT_TYPE(V_028A90_THREAD_TRACE_FINISH) | EVENT_INDEX(0));
+
+   for (unsigned se = 0; se < max_se; se++) {
+      /* Target SEi and SH0. */
+      radeon_set_uconfig_reg(cs, R_030800_GRBM_GFX_INDEX,
+                             S_030800_SE_INDEX(se) |
+                             S_030800_SH_INDEX(0) |
+                             S_030800_INSTANCE_BROADCAST_WRITES(1));
+
+      if (sctx->chip_class == GFX10) {
+         /* Make sure to wait for the trace buffer. */
+         radeon_emit(cs, PKT3(PKT3_WAIT_REG_MEM, 5, 0));
+         radeon_emit(cs, WAIT_REG_MEM_NOT_EQUAL); /* wait until the register is equal to the reference value */
+         radeon_emit(cs, R_008D20_SQ_THREAD_TRACE_STATUS >> 2);  /* register */
+         radeon_emit(cs, 0);
+         radeon_emit(cs, 0); /* reference value */
+         radeon_emit(cs, S_008D20_FINISH_DONE(1)); /* mask */
+         radeon_emit(cs, 4); /* poll interval */
+
+         /* Disable the thread trace mode. */
+         radeon_set_privileged_config_reg(cs, R_008D1C_SQ_THREAD_TRACE_CTRL,
+                                          S_008D1C_MODE(0));
+
+         /* Wait for thread trace completion. */
+         radeon_emit(cs, PKT3(PKT3_WAIT_REG_MEM, 5, 0));
+         radeon_emit(cs, WAIT_REG_MEM_EQUAL); /* wait until the register is equal to the reference value */
+         radeon_emit(cs, R_008D20_SQ_THREAD_TRACE_STATUS >> 2);  /* register */
+         radeon_emit(cs, 0);
+         radeon_emit(cs, 0); /* reference value */
+         radeon_emit(cs, S_008D20_BUSY(1)); /* mask */
+         radeon_emit(cs, 4); /* poll interval */
+      } else {
+         /* Disable the thread trace mode. */
+         radeon_set_uconfig_reg(cs, R_030CD8_SQ_THREAD_TRACE_MODE,
+                                S_030CD8_MODE(0));
+
+         /* Wait for thread trace completion. */
+         radeon_emit(cs, PKT3(PKT3_WAIT_REG_MEM, 5, 0));
+         radeon_emit(cs, WAIT_REG_MEM_EQUAL); /* wait until the register is equal to the reference value */
+         radeon_emit(cs, R_030CE8_SQ_THREAD_TRACE_STATUS >> 2);  /* register */
+         radeon_emit(cs, 0);
+         radeon_emit(cs, 0); /* reference value */
+         radeon_emit(cs, S_030CE8_BUSY(1)); /* mask */
+         radeon_emit(cs, 4); /* poll interval */
+      }
+
+      si_copy_thread_trace_info_regs(sctx, cs, se);
+   }
+
+   /* Restore global broadcasting. */
+   radeon_set_uconfig_reg(cs, R_030800_GRBM_GFX_INDEX,
+                          S_030800_SE_BROADCAST_WRITES(1) |
+                             S_030800_SH_BROADCAST_WRITES(1) |
+                             S_030800_INSTANCE_BROADCAST_WRITES(1));
+}
+
+static void
+si_thread_trace_start(struct si_context *sctx, int family, struct radeon_cmdbuf *cs)
+{
+   struct radeon_winsys *ws = sctx->ws;
+
+   switch (family) {
+      case RING_GFX:
+         radeon_emit(cs, PKT3(PKT3_CONTEXT_CONTROL, 1, 0));
+         radeon_emit(cs, CC0_UPDATE_LOAD_ENABLES(1));
+         radeon_emit(cs, CC1_UPDATE_SHADOW_ENABLES(1));
+         break;
+      case RING_COMPUTE:
+         radeon_emit(cs, PKT3(PKT3_NOP, 0, 0));
+         radeon_emit(cs, 0);
+         break;
+      }
+
+   ws->cs_add_buffer(cs,
+                     sctx->thread_trace->bo,
+                     RADEON_USAGE_READWRITE,
+                     RADEON_DOMAIN_VRAM,
+                     0);
+
+   si_cp_dma_wait_for_idle(sctx, cs);
+
+   /* Make sure to wait-for-idle before starting SQTT. */
+   sctx->flags |=
+      SI_CONTEXT_PS_PARTIAL_FLUSH | SI_CONTEXT_CS_PARTIAL_FLUSH |
+      SI_CONTEXT_INV_ICACHE | SI_CONTEXT_INV_SCACHE | SI_CONTEXT_INV_VCACHE |
+      SI_CONTEXT_INV_L2;
+   sctx->emit_cache_flush(sctx, cs);
+
+   /* Enable SQG events that collects thread trace data. */
+   si_emit_spi_config_cntl(sctx, cs, true);
+
+   si_emit_thread_trace_start(sctx, cs, family);
+}
+
+static void
+si_thread_trace_stop(struct si_context *sctx, int family, struct radeon_cmdbuf *cs)
+{
+   struct radeon_winsys *ws = sctx->ws;
+   switch (family) {
+      case RING_GFX:
+         radeon_emit(sctx->thread_trace->stop_cs[family], PKT3(PKT3_CONTEXT_CONTROL, 1, 0));
+         radeon_emit(sctx->thread_trace->stop_cs[family], CC0_UPDATE_LOAD_ENABLES(1));
+         radeon_emit(sctx->thread_trace->stop_cs[family], CC1_UPDATE_SHADOW_ENABLES(1));
+         break;
+      case RING_COMPUTE:
+         radeon_emit(sctx->thread_trace->stop_cs[family], PKT3(PKT3_NOP, 0, 0));
+         radeon_emit(sctx->thread_trace->stop_cs[family], 0);
+         break;
+   }
+   ws->cs_add_buffer(cs,
+                     sctx->thread_trace->bo,
+                     RADEON_USAGE_READWRITE,
+                     RADEON_DOMAIN_VRAM,
+                     0);
+
+   si_cp_dma_wait_for_idle(sctx, cs);
+
+   /* Make sure to wait-for-idle before stopping SQTT. */
+   sctx->flags |=
+      SI_CONTEXT_PS_PARTIAL_FLUSH | SI_CONTEXT_CS_PARTIAL_FLUSH |
+      SI_CONTEXT_INV_ICACHE | SI_CONTEXT_INV_SCACHE | SI_CONTEXT_INV_VCACHE |
+      SI_CONTEXT_INV_L2;
+   sctx->emit_cache_flush(sctx, cs);
+
+   si_emit_thread_trace_stop(sctx, cs, family);
+
+   /* Restore previous state by disabling SQG events. */
+   si_emit_spi_config_cntl(sctx, cs, false);
+}
+
+
+static void
+si_thread_trace_init_cs(struct si_context *sctx)
+{
+   struct radeon_winsys *ws = sctx->ws;
+
+   /* Thread trace start CS (only handles RING_GFX). */
+   sctx->thread_trace->start_cs[RING_GFX] = CALLOC_STRUCT(radeon_cmdbuf);
+   if (!ws->cs_create(sctx->thread_trace->start_cs[RING_GFX],
+                      sctx->ctx, RING_GFX, NULL, NULL, 0)) {
+      free(sctx->thread_trace->start_cs[RING_GFX]);
+      sctx->thread_trace->start_cs[RING_GFX] = NULL;
+      return;
+   }
+
+   si_thread_trace_start(sctx, RING_GFX, sctx->thread_trace->start_cs[RING_GFX]);
+
+   /* Thread trace stop CS. */
+   sctx->thread_trace->stop_cs[RING_GFX] = CALLOC_STRUCT(radeon_cmdbuf);
+   if (!ws->cs_create(sctx->thread_trace->stop_cs[RING_GFX],
+                      sctx->ctx, RING_GFX, NULL, NULL, 0)) {
+      free(sctx->thread_trace->start_cs[RING_GFX]);
+      sctx->thread_trace->start_cs[RING_GFX] = NULL;
+      free(sctx->thread_trace->stop_cs[RING_GFX]);
+      sctx->thread_trace->stop_cs[RING_GFX] = NULL;
+      return;
+   }
+
+   si_thread_trace_stop(sctx, RING_GFX, sctx->thread_trace->stop_cs[RING_GFX]);
+}
+
+static void
+si_begin_thread_trace(struct si_context *sctx, struct radeon_cmdbuf *rcs)
+{
+   struct radeon_cmdbuf *cs = sctx->thread_trace->start_cs[RING_GFX];
+   sctx->ws->cs_flush(cs, 0, NULL);
+}
+
+static void
+si_end_thread_trace(struct si_context *sctx, struct radeon_cmdbuf *rcs)
+{
+   struct radeon_cmdbuf *cs = sctx->thread_trace->stop_cs[RING_GFX];
+   sctx->ws->cs_flush(cs, 0, &sctx->last_sqtt_fence);
+}
+
+static bool
+si_get_thread_trace(struct si_context *sctx,
+                    struct ac_thread_trace *thread_trace)
+{
+   unsigned max_se = sctx->screen->info.max_se;
+
+   memset(thread_trace, 0, sizeof(*thread_trace));
+   thread_trace->num_traces = max_se;
+
+   sctx->thread_trace->ptr = sctx->ws->buffer_map(sctx->thread_trace->bo,
+                                                          NULL,
+                                                          PIPE_MAP_READ);
+
+   if (!sctx->thread_trace->ptr)
+      return false;
+
+   void *thread_trace_ptr = sctx->thread_trace->ptr;
+
+   for (unsigned se = 0; se < max_se; se++) {
+      uint64_t info_offset = ac_thread_trace_get_info_offset(se);
+      uint64_t data_offset = ac_thread_trace_get_data_offset(sctx->thread_trace, se);
+      void *info_ptr = thread_trace_ptr + info_offset;
+      void *data_ptr = thread_trace_ptr + data_offset;
+      struct ac_thread_trace_info *info =
+         (struct ac_thread_trace_info *)info_ptr;
+
+      struct ac_thread_trace_se thread_trace_se = {0};
+
+      if (!ac_is_thread_trace_complete(&sctx->screen->info, info)) {
+         uint32_t expected_size =
+            ac_get_expected_buffer_size(&sctx->screen->info, info);
+         uint32_t available_size = (info->cur_offset * 32) / 1024;
+
+         fprintf(stderr, "Failed to get the thread trace "
+                 "because the buffer is too small. The "
+                 "hardware needs %d KB but the "
+                 "buffer size is %d KB.\n",
+                 expected_size, available_size);
+         fprintf(stderr, "Please update the buffer size with "
+                 "AMD_THREAD_TRACE_BUFFER_SIZE=<size_in_kbytes>\n");
+         return false;
+      }
+
+      thread_trace_se.data_ptr = data_ptr;
+      thread_trace_se.info = *info;
+      thread_trace_se.shader_engine = se;
+      thread_trace_se.compute_unit = 0;
+
+      thread_trace->traces[se] = thread_trace_se;
+   }
+
+   return true;
+}
+
+
+bool
+si_init_thread_trace(struct si_context *sctx)
+{
+   static bool warn_once = true;
+   if (warn_once) {
+      fprintf(stderr, "*************************************************\n");
+      fprintf(stderr, "* WARNING: Thread trace support is experimental *\n");
+      fprintf(stderr, "*************************************************\n");
+      warn_once = false;
+   }
+
+   sctx->thread_trace = CALLOC_STRUCT(ac_thread_trace_data);
+
+   if (sctx->chip_class < GFX8) {
+      fprintf(stderr, "GPU hardware not supported: refer to "
+              "the RGP documentation for the list of "
+              "supported GPUs!\n");
+      return false;
+   }
+
+   if (sctx->chip_class > GFX10) {
+      fprintf(stderr, "radeonsi: Thread trace is not supported "
+              "for that GPU!\n");
+      return false;
+   }
+
+   /* Default buffer size set to 1MB per SE. */
+   sctx->thread_trace->buffer_size = debug_get_num_option("AMD_THREAD_TRACE_BUFFER_SIZE", 1024) * 1024;
+   sctx->thread_trace->start_frame = 10;
+
+   const char *trigger_file = getenv("AMD_THREAD_TRACE_TRIGGER");
+   if (trigger_file) {
+      sctx->thread_trace->trigger_file = strdup(trigger_file);
+      sctx->thread_trace->start_frame = -1;
+   }
+
+   if (!si_thread_trace_init_bo(sctx))
+      return false;
+
+   si_thread_trace_init_cs(sctx);
+
+   return true;
+}
+
+void
+si_destroy_thread_trace(struct si_context *sctx)
+{
+  struct si_screen *sscreen = sctx->screen;
+   struct pb_buffer *bo = sctx->thread_trace->bo;
+   pb_reference(&bo, NULL);
+
+   if (sctx->thread_trace->trigger_file)
+      free(sctx->thread_trace->trigger_file);
+   sscreen->ws->cs_destroy(sctx->thread_trace->start_cs[RING_GFX]);
+   sscreen->ws->cs_destroy(sctx->thread_trace->stop_cs[RING_GFX]);
+   free(sctx->thread_trace);
+   sctx->thread_trace = NULL;
+}
+
+static uint64_t num_frames = 0;
+
+void
+si_handle_thread_trace(struct si_context *sctx, struct radeon_cmdbuf *rcs)
+{
+   /* Should we enable SQTT yet? */
+   if (!sctx->thread_trace_enabled) {
+      bool frame_trigger = num_frames == sctx->thread_trace->start_frame;
+      bool file_trigger = false;
+      if (sctx->thread_trace->trigger_file &&
+          access(sctx->thread_trace->trigger_file, W_OK) == 0) {
+         if (unlink(sctx->thread_trace->trigger_file) == 0) {
+            file_trigger = true;
+         } else {
+            /* Do not enable tracing if we cannot remove the file,
+             * because by then we'll trace every frame.
+             */
+            fprintf(stderr, "radeonsi: could not remove thread trace trigger file, ignoring\n");
+         }
+      }
+
+      if (frame_trigger || file_trigger) {
+         /* Wait for last submission */
+         sctx->ws->fence_wait(sctx->ws, sctx->last_gfx_fence, PIPE_TIMEOUT_INFINITE);
+
+         /* Start SQTT */
+         si_begin_thread_trace(sctx, rcs);
+
+         sctx->thread_trace_enabled = true;
+         sctx->thread_trace->start_frame = -1;
+      }
+   } else {
+      struct ac_thread_trace thread_trace = {0};
+
+      /* Stop SQTT */
+      si_end_thread_trace(sctx, rcs);
+      sctx->thread_trace_enabled = false;
+      sctx->thread_trace->start_frame = -1;
+      assert (sctx->last_sqtt_fence);
+
+      /* Wait for SQTT to finish and read back the bo */
+      if (sctx->ws->fence_wait(sctx->ws, sctx->last_sqtt_fence, PIPE_TIMEOUT_INFINITE) &&
+          si_get_thread_trace(sctx, &thread_trace)) {
+         ac_dump_thread_trace(&sctx->screen->info, &thread_trace);
+      } else {
+         fprintf(stderr, "Failed to read the trace\n");
+      }
+   }
+
+   num_frames++;
+}
+
+
+static void
+si_emit_thread_trace_userdata(struct si_context* sctx,
+                              struct radeon_cmdbuf *cs,
+                              const void *data, uint32_t num_dwords)
+{
+   const uint32_t *dwords = (uint32_t *)data;
+
+   while (num_dwords > 0) {
+      uint32_t count = MIN2(num_dwords, 2);
+
+      /* Without the perfctr bit the CP might not always pass the
+       * write on correctly. */
+      radeon_set_uconfig_reg_seq(cs, R_030D08_SQ_THREAD_TRACE_USERDATA_2, count, sctx->chip_class >= GFX10);
+
+      radeon_emit_array(cs, dwords, count);
+
+      dwords += count;
+      num_dwords -= count;
+   }
+}
+
+static void
+si_emit_spi_config_cntl(struct si_context* sctx,
+           struct radeon_cmdbuf *cs, bool enable)
+{
+   if (sctx->chip_class >= GFX9) {
+      uint32_t spi_config_cntl = S_031100_GPR_WRITE_PRIORITY(0x2c688) |
+                                 S_031100_EXP_PRIORITY_ORDER(3) |
+                                 S_031100_ENABLE_SQG_TOP_EVENTS(enable) |
+                                 S_031100_ENABLE_SQG_BOP_EVENTS(enable);
+
+      if (sctx->chip_class == GFX10)
+         spi_config_cntl |= S_031100_PS_PKR_PRIORITY_CNTL(3);
+
+      radeon_set_uconfig_reg(cs, R_031100_SPI_CONFIG_CNTL, spi_config_cntl);
+   } else {
+      /* SPI_CONFIG_CNTL is a protected register on GFX6-GFX8. */
+      radeon_set_privileged_config_reg(cs, R_009100_SPI_CONFIG_CNTL,
+                                       S_009100_ENABLE_SQG_TOP_EVENTS(enable) |
+                                       S_009100_ENABLE_SQG_BOP_EVENTS(enable));
+   }
+}
+
+void
+si_sqtt_write_event_marker(struct si_context* sctx, struct radeon_cmdbuf *rcs,
+                           enum rgp_sqtt_marker_event_type api_type,
+                           uint32_t vertex_offset_user_data,
+                           uint32_t instance_offset_user_data,
+                           uint32_t draw_index_user_data)
+{
+   static uint32_t num_events = 0;
+   struct rgp_sqtt_marker_event marker = {0};
+
+   marker.identifier = RGP_SQTT_MARKER_IDENTIFIER_EVENT;
+   marker.api_type = api_type;
+   marker.cmd_id = num_events++;
+   marker.cb_id = 0;
+
+   if (vertex_offset_user_data == UINT_MAX ||
+       instance_offset_user_data == UINT_MAX) {
+      vertex_offset_user_data = 0;
+      instance_offset_user_data = 0;
+   }
+
+   if (draw_index_user_data == UINT_MAX)
+      draw_index_user_data = vertex_offset_user_data;
+
+   marker.vertex_offset_reg_idx = vertex_offset_user_data;
+   marker.instance_offset_reg_idx = instance_offset_user_data;
+   marker.draw_index_reg_idx = draw_index_user_data;
+
+   si_emit_thread_trace_userdata(sctx, rcs, &marker, sizeof(marker) / 4);
+}
diff --git a/src/gallium/drivers/radeonsi/si_state_draw.cpp b/src/gallium/drivers/radeonsi/si_state_draw.cpp
index e942c3cabb3..c4042b53533 100644
--- a/src/gallium/drivers/radeonsi/si_state_draw.cpp
+++ b/src/gallium/drivers/radeonsi/si_state_draw.cpp
@@ -23,6 +23,7 @@
  */
 
 #include "ac_debug.h"
+#include "ac_sqtt.h"
 #include "si_build_pm4.h"
 #include "sid.h"
 #include "util/u_index_modify.h"
@@ -815,6 +816,15 @@ static void si_emit_draw_registers(struct si_context *sctx, const struct pipe_dr
    }
 }
 
+#define EMIT_SQTT_END_DRAW do {                                          \
+      if (GFX_VERSION >= GFX9 && unlikely(sctx->thread_trace_enabled)) { \
+         radeon_emit(&sctx->gfx_cs, PKT3(PKT3_EVENT_WRITE, 0, 0));       \
+         radeon_emit(&sctx->gfx_cs,                                      \
+                     EVENT_TYPE(V_028A90_THREAD_TRACE_MARKER) |          \
+                     EVENT_INDEX(0));                                    \
+      }                                                                  \
+   } while (0)
+
 template <chip_class GFX_VERSION, si_has_ngg NGG, si_has_prim_discard_cs ALLOW_PRIM_DISCARD_CS>
 static void si_emit_draw_packets(struct si_context *sctx, const struct pipe_draw_info *info,
                                  const struct pipe_draw_indirect_info *indirect,
@@ -831,6 +841,11 @@ static void si_emit_draw_packets(struct si_context *sctx, const struct pipe_draw
    uint32_t use_opaque = 0;
    uint64_t index_va = 0;
 
+   if (unlikely(sctx->thread_trace_enabled)) {
+      si_sqtt_write_event_marker(sctx, &sctx->gfx_cs, EventCmdDraw,
+                                 UINT_MAX, UINT_MAX, UINT_MAX);
+   }
+
    if (indirect && indirect->count_from_stream_output) {
       struct si_streamout_target *t = (struct si_streamout_target *)indirect->count_from_stream_output;
 
@@ -1031,6 +1046,7 @@ static void si_emit_draw_packets(struct si_context *sctx, const struct pipe_draw
                                                     original_index_size, base_vertex,
                                                     va, MIN2(index_max_size, draws[i].count));
             }
+            EMIT_SQTT_END_DRAW;
             return;
          }
 
@@ -1088,6 +1104,7 @@ static void si_emit_draw_packets(struct si_context *sctx, const struct pipe_draw
                radeon_emit(cs, draws[i].count);
                radeon_emit(cs, V_0287F0_DI_SRC_SEL_AUTO_INDEX);
             }
+            EMIT_SQTT_END_DRAW;
             return;
          }
 
@@ -1114,6 +1131,8 @@ static void si_emit_draw_packets(struct si_context *sctx, const struct pipe_draw
             sctx->last_base_vertex = draws[num_draws - 1].start;
       }
    }
+
+   EMIT_SQTT_END_DRAW;
 }
 
 extern "C"



More information about the mesa-commit mailing list