From 1f464d53010f61ca2ef10e88e4eeefc067f4c8ce Mon Sep 17 00:00:00 2001
From: Rob Clark <robdclark@gmail.com>
Date: Mon, 9 Jul 2018 13:17:12 -0400
Subject: [PATCH] freedreno/ir3: output ir3 and nir asm for frameretrace

See: https://github.com/janesma/apitrace/commit/298dc8195bf082fe1f47aa474e28411f85dd5393

Signed-off-by: Rob Clark <robdclark@gmail.com>
---
 src/gallium/drivers/freedreno/disasm.h        | 43 +++++++++++++++++++
 .../drivers/freedreno/freedreno_screen.c      | 12 ++++++
 .../drivers/freedreno/ir3/ir3_compiler_nir.c  |  6 +++
 .../drivers/freedreno/ir3/ir3_shader.c        |  8 ++++
 4 files changed, 69 insertions(+)

diff --git a/src/gallium/drivers/freedreno/disasm.h b/src/gallium/drivers/freedreno/disasm.h
index 3266c6a484f..fd8053bf49d 100644
--- a/src/gallium/drivers/freedreno/disasm.h
+++ b/src/gallium/drivers/freedreno/disasm.h
@@ -25,6 +25,17 @@
 #define DISASM_H_
 
 #include <stdio.h>
+#include <stdbool.h>
+
+#include "util/u_debug.h"
+
+enum fd_shader_debug {
+	FD_DBG_SHADER_VS = 0x01,
+	FD_DBG_SHADER_FS = 0x02,
+	FD_DBG_SHADER_CS = 0x04,
+};
+
+extern enum fd_shader_debug fd_shader_debug;
 
 enum shader_t {
 	SHADER_VERTEX,
@@ -36,6 +47,38 @@ enum shader_t {
 	SHADER_MAX,
 };
 
+static inline bool
+shader_debug_enabled(enum shader_t type)
+{
+	switch (type) {
+	case SHADER_VERTEX:      return !!(fd_shader_debug & FD_DBG_SHADER_VS);
+	case SHADER_FRAGMENT:    return !!(fd_shader_debug & FD_DBG_SHADER_FS);
+	case SHADER_COMPUTE:     return !!(fd_shader_debug & FD_DBG_SHADER_CS);
+	default:
+		debug_assert(0);
+		return false;
+	}
+}
+
+static inline const char *
+shader_stage_name(enum shader_t type)
+{
+	/* NOTE these names are chosen to match the INTEL_DEBUG output
+	 * which frameretrace parses.  Hurray accidental ABI!
+	 */
+	switch (type) {
+	case SHADER_VERTEX:      return "vertex";
+	case SHADER_TCS:         return "tessellation control";
+	case SHADER_TES:         return "tessellation evaluation";
+	case SHADER_GEOM:        return "geometry";
+	case SHADER_FRAGMENT:    return "fragment";
+	case SHADER_COMPUTE:     return "compute";
+	default:
+		debug_assert(0);
+		return NULL;
+	}
+}
+
 /* bitmask of debug flags */
 enum debug_t {
 	PRINT_RAW      = 0x1,    /* dump raw hexdump */
diff --git a/src/gallium/drivers/freedreno/freedreno_screen.c b/src/gallium/drivers/freedreno/freedreno_screen.c
index 1baebd71789..be5b88b0dd6 100644
--- a/src/gallium/drivers/freedreno/freedreno_screen.c
+++ b/src/gallium/drivers/freedreno/freedreno_screen.c
@@ -93,6 +93,17 @@ int fd_mesa_debug = 0;
 bool fd_binning_enabled = true;
 static bool glsl120 = false;
 
+static const struct debug_named_value shader_debug_options[] = {
+		{"vs", FD_DBG_SHADER_VS, "Print shader disasm for vertex shaders"},
+		{"fs", FD_DBG_SHADER_FS, "Print shader disasm for fragment shaders"},
+		{"cs", FD_DBG_SHADER_CS, "Print shader disasm for compute shaders"},
+		DEBUG_NAMED_VALUE_END
+};
+
+DEBUG_GET_ONCE_FLAGS_OPTION(fd_shader_debug, "FD_SHADER_DEBUG", shader_debug_options, 0)
+
+enum fd_shader_debug fd_shader_debug = 0;
+
 static const char *
 fd_screen_get_name(struct pipe_screen *pscreen)
 {
@@ -783,6 +794,7 @@ fd_screen_create(struct fd_device *dev)
 	uint64_t val;
 
 	fd_mesa_debug = debug_get_option_fd_mesa_debug();
+	fd_shader_debug = debug_get_option_fd_shader_debug();
 
 	if (fd_mesa_debug & FD_DBG_NOBIN)
 		fd_binning_enabled = false;
diff --git a/src/gallium/drivers/freedreno/ir3/ir3_compiler_nir.c b/src/gallium/drivers/freedreno/ir3/ir3_compiler_nir.c
index 9f74fa2530a..d8414903e75 100644
--- a/src/gallium/drivers/freedreno/ir3/ir3_compiler_nir.c
+++ b/src/gallium/drivers/freedreno/ir3/ir3_compiler_nir.c
@@ -220,6 +220,12 @@ compile_init(struct ir3_compiler *compiler,
 		nir_print_shader(ctx->s, stdout);
 	}
 
+	if (shader_debug_enabled(so->type)) {
+		fprintf(stderr, "NIR (final form) for %s shader:\n",
+			shader_stage_name(so->type));
+		nir_print_shader(ctx->s, stderr);
+	}
+
 	ir3_nir_scan_driver_consts(ctx->s, &so->const_layout);
 
 	so->num_uniforms = ctx->s->num_uniforms;
diff --git a/src/gallium/drivers/freedreno/ir3/ir3_shader.c b/src/gallium/drivers/freedreno/ir3/ir3_shader.c
index 15bda50b25b..b0663d5c5ca 100644
--- a/src/gallium/drivers/freedreno/ir3/ir3_shader.c
+++ b/src/gallium/drivers/freedreno/ir3/ir3_shader.c
@@ -162,6 +162,14 @@ assemble_variant(struct ir3_shader_variant *v)
 		ir3_shader_disasm(v, bin, stdout);
 	}
 
+	if (shader_debug_enabled(v->shader->type)) {
+		fprintf(stderr, "Native code for unnamed %s shader %s:\n",
+			shader_stage_name(v->shader->type), v->shader->nir->info.name);
+		if (v->shader->type == SHADER_FRAGMENT)
+			fprintf(stderr, "SIMD0\n");
+		ir3_shader_disasm(v, bin, stderr);
+	}
+
 	free(bin);
 
 	/* no need to keep the ir around beyond this point: */
-- 
2.30.2