src/panfrost/bifrost/bi_print.c

   1 /*
   2  * Copyright (C) 2019 Connor Abbott <cwabbott0@gmail.com>
   3  * Copyright (C) 2019 Lyude Paul <thatslyude@gmail.com>
   4  * Copyright (C) 2019 Ryan Houdek <Sonicadvance1@gmail.com>
   5  * Copyright (C) 2019-2020 Collabora, Ltd.
   6  *
   7  * Permission is hereby granted, free of charge, to any person obtaining a
   8  * copy of this software and associated documentation files (the "Software"),
   9  * to deal in the Software without restriction, including without limitation
  10  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  11  * and/or sell copies of the Software, and to permit persons to whom the
  12  * Software is furnished to do so, subject to the following conditions:
  13  *
  14  * The above copyright notice and this permission notice (including the next
  15  * paragraph) shall be included in all copies or substantial portions of the
  16  * Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  21  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  22  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  23  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  24  * SOFTWARE.
  25  */
  26
  27 #include "bi_print.h"
  28
  29 const char *
  30 bi_output_mod_name(enum bifrost_outmod mod)
  31 {
  32         switch (mod) {
  33         case BIFROST_NONE: return "";
  34         case BIFROST_POS: return ".pos";
  35         case BIFROST_SAT_SIGNED: return ".sat_signed";
  36         case BIFROST_SAT: return ".sat";
  37         default: return "invalid";
  38         }
  39 }
  40
  41 const char *
  42 bi_minmax_mode_name(enum bifrost_minmax_mode mod)
  43 {
  44         switch (mod) {
  45         case BIFROST_MINMAX_NONE: return "";
  46         case BIFROST_NAN_WINS: return ".nan_wins";
  47         case BIFROST_SRC1_WINS: return ".src1_wins";
  48         case BIFROST_SRC0_WINS: return ".src0_wins";
  49         default: return "invalid";
  50         }
  51 }
  52
  53 const char *
  54 bi_round_mode_name(enum bifrost_roundmode mod)
  55 {
  56         switch (mod) {
  57         case BIFROST_RTE: return "";
  58         case BIFROST_RTP: return ".rtp";
  59         case BIFROST_RTN: return ".rtn";
  60         case BIFROST_RTZ: return ".rtz";
  61         default: return "invalid";
  62         }
  63 }
  64
  65 const char *
  66 bi_csel_cond_name(enum bifrost_csel_cond cond)
  67 {
  68         switch (cond) {
  69         case BIFROST_FEQ_F: return "feq.f";
  70         case BIFROST_FGT_F: return "fgt.f";
  71         case BIFROST_FGE_F: return "fge.f";
  72         case BIFROST_IEQ_F: return "ieq.f";
  73         case BIFROST_IGT_I: return "igt.i";
  74         case BIFROST_IGE_I: return "uge.i";
  75         case BIFROST_UGT_I: return "ugt.i";
  76         case BIFROST_UGE_I: return "uge.i";
  77         default: return "invalid";
  78         }
  79 }
  80
  81 const char *
  82 bi_interp_mode_name(enum bifrost_interp_mode mode)
  83 {
  84         switch (mode) {
  85         case BIFROST_INTERP_PER_FRAG: return ".per_frag";
  86         case BIFROST_INTERP_CENTROID: return ".centroid";
  87         case BIFROST_INTERP_DEFAULT: return "";
  88         case BIFROST_INTERP_EXPLICIT: return ".explicit";
  89         default: return ".unknown";
  90         }
  91 }
  92
  93 const char *
  94 bi_ldst_type_name(enum bifrost_ldst_type type)
  95 {
  96         switch (type) {
  97         case BIFROST_LDST_F16: return "f16";
  98         case BIFROST_LDST_F32: return "f32";
  99         case BIFROST_LDST_I32: return "i32";
 100         case BIFROST_LDST_U32: return "u32";
 101         default: return "invalid";
 102         }
 103 }
 104
 105 /* The remaining functions in this file are for IR-internal
 106  * structures; the disassembler doesn't use them */
 107
 108 static const char *
 109 bi_class_name(enum bi_class cl)
 110 {
 111         switch (cl) {
 112         case BI_ADD: return "add";
 113         case BI_ATEST: return "atest";
 114         case BI_BRANCH: return "branch";
 115         case BI_CMP: return "cmp";
 116         case BI_BLEND: return "blend";
 117         case BI_BITWISE: return "bitwise";
 118         case BI_CONVERT: return "convert";
 119         case BI_CSEL: return "csel";
 120         case BI_DISCARD: return "discard";
 121         case BI_FMA: return "fma";
 122         case BI_FREXP: return "frexp";
 123         case BI_LOAD: return "load";
 124         case BI_LOAD_UNIFORM: return "load_uniform";
 125         case BI_LOAD_ATTR: return "load_attr";
 126         case BI_LOAD_VAR: return "load_var";
 127         case BI_LOAD_VAR_ADDRESS: return "load_var_address";
 128         case BI_MINMAX: return "minmax";
 129         case BI_MOV: return "mov";
 130         case BI_SHIFT: return "shift";
 131         case BI_STORE: return "store";
 132         case BI_STORE_VAR: return "store_var";
 133         case BI_SPECIAL: return "special";
 134         case BI_SWIZZLE: return "swizzle";
 135         case BI_TEX: return "tex";
 136         case BI_ROUND: return "round";
 137         default: return "unknown_class";
 138         }
 139 }
 140
 141 static void
 142 bi_print_index(FILE *fp, bi_instruction *ins, unsigned index)
 143 {
 144         if (!index)
 145                 fprintf(fp, "_");
 146         else if (index & BIR_INDEX_REGISTER)
 147                 fprintf(fp, "br%u", index & ~BIR_INDEX_REGISTER);
 148         else if (index & BIR_INDEX_UNIFORM)
 149                 fprintf(fp, "u%u", index & ~BIR_INDEX_UNIFORM);
 150         else if (index & BIR_INDEX_CONSTANT)
 151                 fprintf(fp, "#0x%" PRIx64, ins->constant.u64);
 152         else if (index & BIR_INDEX_ZERO)
 153                 fprintf(fp, "#0");
 154         else if (index & BIR_IS_REG)
 155                 fprintf(fp, "r%u", index >> 1);
 156         else
 157                 fprintf(fp, "%u", (index >> 1) - 1);
 158 }
 159
 160 static void
 161 bi_print_src(FILE *fp, bi_instruction *ins, unsigned s)
 162 {
 163         unsigned src = ins->src[s];
 164         bool mods = bi_has_source_mods(ins);
 165         bool abs = ins->src_abs[s] && mods;
 166         bool neg = ins->src_neg[s] && mods;
 167
 168         if (neg)
 169                 fprintf(fp, "-");
 170
 171         if (abs)
 172                 fprintf(fp, "abs(");
 173
 174         bi_print_index(fp, ins, src);
 175
 176         if (abs)
 177                 fprintf(fp, ")");
 178 }
 179
 180 /* Prints a NIR ALU type in Bifrost-style ".f32" ".i8" etc */
 181
 182 static void
 183 bi_print_alu_type(nir_alu_type t, FILE *fp)
 184 {
 185         unsigned size = nir_alu_type_get_type_size(t);
 186         nir_alu_type base = nir_alu_type_get_base_type(t);
 187
 188         switch (base) {
 189         case nir_type_int:
 190                 fprintf(fp, ".i");
 191                 break;
 192         case nir_type_uint:
 193                 fprintf(fp, ".u");
 194                 break;
 195         case nir_type_bool:
 196                 fprintf(fp, ".b");
 197                 break;
 198         case nir_type_float:
 199                 fprintf(fp, ".f");
 200                 break;
 201         default:
 202                 fprintf(fp, ".unknown");
 203                 break;
 204         }
 205
 206         fprintf(fp, "%u", size);
 207 }
 208
 209 static void
 210 bi_print_swizzle(bi_instruction *ins, unsigned src, FILE *fp)
 211 {
 212         unsigned size = MAX2(nir_alu_type_get_type_size(ins->dest_type), 8);
 213         unsigned count = (size == 64) ? 1 : (32 / size);
 214
 215         fprintf(fp, ".");
 216
 217         for (unsigned u = 0; u < count; ++u) {
 218                 assert(ins->swizzle[src][u] < 4);
 219                 fputc("xyzw"[ins->swizzle[src][u]], fp);
 220         }
 221 }
 222
 223 static const char *
 224 bi_bitwise_op_name(enum bi_bitwise_op op)
 225 {
 226         switch (op) {
 227         case BI_BITWISE_AND: return "and";
 228         case BI_BITWISE_OR: return "or";
 229         case BI_BITWISE_XOR: return "xor";
 230         default: return "invalid";
 231         }
 232 }
 233
 234 static const char *
 235 bi_special_op_name(enum bi_special_op op)
 236 {
 237         switch (op) {
 238         case BI_SPECIAL_FRCP: return "frcp";
 239         case BI_SPECIAL_FRSQ: return "frsq";
 240         case BI_SPECIAL_FATAN: return "fatan";
 241         case BI_SPECIAL_FSIN: return "fsin";
 242         case BI_SPECIAL_FCOS: return "fcos";
 243         case BI_SPECIAL_FEXP: return "fexp";
 244         case BI_SPECIAL_FLOG2: return "flog2";
 245         case BI_SPECIAL_FLOGE: return "flog";
 246         default: return "invalid";
 247         }
 248 }
 249
 250 static void
 251 bi_print_load_vary(struct bi_load_vary *load, FILE *fp)
 252 {
 253         fprintf(fp, "%s", bi_interp_mode_name(load->interp_mode));
 254
 255         if (load->reuse)
 256                 fprintf(fp, ".reuse");
 257
 258         if (load->flat)
 259                 fprintf(fp, ".flat");
 260 }
 261
 262 static const char *
 263 bi_cond_name(enum bi_cond cond)
 264 {
 265         switch (cond) {
 266         case BI_COND_ALWAYS: return ".always";
 267         case BI_COND_LT: return ".lt";
 268         case BI_COND_LE: return ".le";
 269         case BI_COND_GE: return ".ge";
 270         case BI_COND_GT: return ".gt";
 271         case BI_COND_EQ: return ".eq";
 272         case BI_COND_NE: return ".ne";
 273         default: return "invalid";
 274         }
 275 }
 276
 277 static void
 278 bi_print_branch(struct bi_branch *branch, FILE *fp)
 279 {
 280         fprintf(fp, "%s", bi_cond_name(branch->cond));
 281 }
 282
 283 static void
 284 bi_print_writemask(bi_instruction *ins, FILE *fp)
 285 {
 286         unsigned bits_per_comp = nir_alu_type_get_type_size(ins->dest_type);
 287         assert(bits_per_comp);
 288         unsigned bytes_per_comp = MAX2(bits_per_comp / 8, 1);
 289         unsigned comps = 16 / bytes_per_comp;
 290         unsigned smask = (1 << bytes_per_comp) - 1;
 291         fprintf(fp, ".");
 292
 293         for (unsigned i = 0; i < comps; ++i) {
 294                 unsigned masked = (ins->writemask >> (i * bytes_per_comp)) & smask;
 295                 if (!masked)
 296                         continue;
 297
 298                 assert(masked == smask);
 299                 assert(i < 4);
 300                 fputc("xyzw"[i], fp);
 301         }
 302 }
 303
 304 void
 305 bi_print_instruction(bi_instruction *ins, FILE *fp)
 306 {
 307         if (ins->type == BI_MINMAX)
 308                 fprintf(fp, "%s", ins->op.minmax == BI_MINMAX_MIN ? "min" : "max");
 309         else if (ins->type == BI_BITWISE)
 310                 fprintf(fp, "%s", bi_bitwise_op_name(ins->op.bitwise));
 311         else if (ins->type == BI_ROUND)
 312                 fprintf(fp, ins->op.round == BI_ROUND_MODE ? "roundMode": "round");
 313         else if (ins->type == BI_SPECIAL)
 314                 fprintf(fp, "%s", bi_special_op_name(ins->op.special));
 315         else
 316                 fprintf(fp, "%s", bi_class_name(ins->type));
 317
 318         if (ins->type == BI_MINMAX)
 319                 fprintf(fp, "%s", bi_minmax_mode_name(ins->minmax));
 320         else if (ins->type == BI_LOAD_VAR)
 321                 bi_print_load_vary(&ins->load_vary, fp);
 322         else if (ins->type == BI_BRANCH)
 323                 bi_print_branch(&ins->branch, fp);
 324         else if (ins->type == BI_CSEL)
 325                 fprintf(fp, "%s", bi_cond_name(ins->csel_cond));
 326         else if (ins->type == BI_BLEND)
 327                 fprintf(fp, ".loc%u", ins->blend_location);
 328
 329         if (ins->dest)
 330                 bi_print_alu_type(ins->dest_type, fp);
 331
 332         if (bi_has_outmod(ins))
 333                 fprintf(fp, "%s", bi_output_mod_name(ins->outmod));
 334
 335         if (bi_class_props[ins->type] & BI_ROUNDMODE)
 336                 fprintf(fp, "%s", bi_round_mode_name(ins->roundmode));
 337
 338         fprintf(fp, " ");
 339         bi_print_index(fp, ins, ins->dest);
 340
 341         if (ins->dest)
 342                 bi_print_writemask(ins, fp);
 343
 344         fprintf(fp, ", ");
 345
 346         bi_foreach_src(ins, s) {
 347                 bi_print_src(fp, ins, s);
 348
 349                 if (ins->src[s] && !(ins->src[s] & (BIR_INDEX_CONSTANT | BIR_INDEX_ZERO)))
 350                         bi_print_swizzle(ins, s, fp);
 351
 352                 bool is_convert = ins->type == BI_CONVERT && s == 0;
 353                 bool is_branch = ins->type == BI_BRANCH && s < 2 && ins->branch.cond != BI_COND_ALWAYS;
 354                 bool is_csel = ins->type == BI_CSEL && s < 2;
 355
 356                 if (is_convert || is_branch || is_csel)
 357                         bi_print_alu_type(ins->src_types[s], fp);
 358
 359                 if (s < BIR_SRC_COUNT)
 360                         fprintf(fp, ", ");
 361         }
 362
 363         if (ins->type == BI_BRANCH) {
 364                 if (ins->branch.target)
 365                         fprintf(fp, "-> block%u", ins->branch.target->name);
 366                 else
 367                         fprintf(fp, "-> blockhole");
 368         }
 369
 370         fprintf(fp, "\n");
 371 }
 372
 373 void
 374 bi_print_bundle(bi_bundle *bundle, FILE *fp)
 375 {
 376         bi_instruction *ins[2] = { bundle->fma, bundle->add };
 377
 378         for (unsigned i = 0; i < 2; ++i) {
 379                 if (ins[i])
 380                         bi_print_instruction(ins[i], fp);
 381                 else
 382                         fprintf(fp, "nop\n");
 383         }
 384 }
 385
 386 void
 387 bi_print_clause(bi_clause *clause, FILE *fp)
 388 {
 389         fprintf(fp, "\tid(%u)", clause->scoreboard_id);
 390
 391         if (clause->dependencies) {
 392                 fprintf(fp, ", wait(");
 393
 394                 for (unsigned i = 0; i < 8; ++i) {
 395                         if (clause->dependencies & (1 << i))
 396                                 fprintf(fp, "%u ", i);
 397                 }
 398
 399                 fprintf(fp, ")");
 400         }
 401
 402         if (!clause->back_to_back)
 403                 fprintf(fp, " nbb %s", clause->branch_conditional ? "branch-cond" : "branch-uncond");
 404
 405         if (clause->data_register_write_barrier)
 406                 fprintf(fp, " drwb");
 407
 408         fprintf(fp, "\n");
 409
 410         if (clause->instruction_count) {
 411                 assert(!clause->bundle_count);
 412
 413                 for (unsigned i = 0; i < clause->instruction_count; ++i)
 414                         bi_print_instruction(clause->instructions[i], fp);
 415         } else {
 416                 assert(clause->bundle_count);
 417
 418                 for (unsigned i = 0; i < clause->bundle_count; ++i)
 419                         bi_print_bundle(&clause->bundles[i], fp);
 420         }
 421
 422         if (clause->constant_count) {
 423                 for (unsigned i = 0; i < clause->constant_count; ++i)
 424                         fprintf(fp, "%" PRIx64 " ", clause->constants[i]);
 425
 426                 fprintf(fp, "\n");
 427         }
 428 }
 429
 430 void
 431 bi_print_block(bi_block *block, FILE *fp)
 432 {
 433         fprintf(fp, "block%u {\n", block->name);
 434
 435         if (block->scheduled) {
 436                 bi_foreach_clause_in_block(block, clause)
 437                         bi_print_clause(clause, fp);
 438         } else {
 439                 bi_foreach_instr_in_block(block, ins)
 440                         bi_print_instruction(ins, fp);
 441         }
 442
 443         fprintf(fp, "}");
 444
 445         if (block->successors[0]) {
 446                 fprintf(fp, " -> ");
 447
 448                 for (unsigned i = 0; i < ARRAY_SIZE(block->successors); ++i) {
 449                         if (block->successors[i])
 450                                 fprintf(fp, "block%u ", block->successors[i]->name);
 451                 }
 452         }
 453
 454         if (block->predecessors->entries) {
 455                 fprintf(fp, " from");
 456
 457                 bi_foreach_predecessor(block, pred)
 458                         fprintf(fp, " block%u", pred->name);
 459         }
 460
 461         fprintf(fp, "\n\n");
 462 }
 463
 464 void
 465 bi_print_shader(bi_context *ctx, FILE *fp)
 466 {
 467         bi_foreach_block(ctx, block)
 468                 bi_print_block(block, fp);
 469 }