agx: Add instruction printing

Signed-off-by: Alyssa Rosenzweig <alyssa@rosenzweig.io> Acked-by: Jason Ekstrand <jason@jlekstrand.net> Acked-by: Bas Nieuwenhuizen <bas@basnieuwenhuizen.nl> Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/10582>
2021-04-11 15:29:08 -04:00
parent 8af9822d14
commit 583684b5a7
1 changed files with 141 additions and 1 deletions
--- a/src/asahi/compiler/agx_print.c
+++ b/src/asahi/compiler/agx_print.c
@@ -24,10 +24,150 @@

 #include "agx_compiler.h"

+static void
+agx_print_sized(char prefix, unsigned value, enum agx_size size, FILE *fp)
+{
+   switch (size) {
+   case AGX_SIZE_16:
+      fprintf(fp, "%c%u%c", prefix, value >> 1, (value & 1) ? 'h' : 'l');
+      return;
+   case AGX_SIZE_32:
+      assert((value & 1) == 0);
+      fprintf(fp, "%c%u", prefix, value >> 1);
+      return;
+   case AGX_SIZE_64:
+      assert((value & 1) == 0);
+      fprintf(fp, "%c%u:%c%u", prefix, value >> 1,
+            prefix, (value >> 1) + 1);
+      return;
+   }
+
+   unreachable("Invalid size");
+}
+
+static void
+agx_print_index(agx_index index, FILE *fp)
+{
+   switch (index.type) {
+   case AGX_INDEX_NULL:
+      fprintf(fp, "_");
+      return;
+
+   case AGX_INDEX_NORMAL:
+      if (index.cache)
+         fprintf(fp, "$");
+
+      if (index.discard)
+         fprintf(fp, "`");
+
+      fprintf(fp, "%u", index.value);
+      break;
+
+   case AGX_INDEX_IMMEDIATE:
+      fprintf(fp, "#%u", index.value);
+      break;
+
+   case AGX_INDEX_UNIFORM:
+      agx_print_sized('u', index.value, index.size, fp);
+      break;
+
+   case AGX_INDEX_REGISTER:
+      agx_print_sized('r', index.value, index.size, fp);
+      break;
+
+   default:
+      unreachable("Invalid index type");
+   }
+
+   /* Print length suffixes if not implied */
+   if (index.type == AGX_INDEX_NORMAL || index.type == AGX_INDEX_IMMEDIATE) {
+      if (index.size == AGX_SIZE_16)
+         fprintf(fp, "h");
+      else if (index.size == AGX_SIZE_64)
+         fprintf(fp, "d");
+   }
+
+   if (index.abs)
+      fprintf(fp, ".abs");
+
+   if (index.neg)
+      fprintf(fp, ".neg");
+}
+
 void
 agx_print_instr(agx_instr *I, FILE *fp)
 {
-   /* Stub */
+   assert(I->op < AGX_NUM_OPCODES);
+   struct agx_opcode_info info = agx_opcodes_info[I->op];
+
+   fprintf(fp, "   %s", info.name);
+
+   if (I->saturate)
+      fprintf(fp, ".sat");
+
+   if (I->last)
+      fprintf(fp, ".last");
+
+   fprintf(fp, " ");
+
+   bool print_comma = false;
+
+   for (unsigned d = 0; d < info.nr_dests; ++d) {
+      if (print_comma)
+         fprintf(fp, ", ");
+      else
+         print_comma = true;
+
+      agx_print_index(I->dest[d], fp);
+   }
+
+   for (unsigned s = 0; s < info.nr_srcs; ++s) {
+      if (print_comma)
+         fprintf(fp, ", ");
+      else
+         print_comma = true;
+
+      agx_print_index(I->src[s], fp);
+   }
+
+   if (I->mask) {
+      fprintf(fp, ", ");
+
+      for (unsigned i = 0; i < 4; ++i) {
+         if (I->mask & (1 << i))
+            fprintf(fp, "%c", "xyzw"[i]);
+      }
+   }
+
+   /* TODO: Do better for enums, truth tables, etc */
+   if (info.immediates) {
+      if (print_comma)
+         fprintf(fp, ", ");
+      else
+         print_comma = true;
+
+      fprintf(fp, "#%X", I->imm);
+   }
+
+   if (info.immediates & AGX_IMMEDIATE_DIM) {
+      if (print_comma)
+         fprintf(fp, ", ");
+      else
+         print_comma = true;
+
+      fprintf(fp, "dim %u", I->dim); // TODO enumify
+   }
+
+   if (info.immediates & AGX_IMMEDIATE_SCOREBOARD) {
+      if (print_comma)
+         fprintf(fp, ", ");
+      else
+         print_comma = true;
+
+      fprintf(fp, "slot %u", I->scoreboard);
+   }
+
+   fprintf(fp, "\n");
 }

 void