src/gallium/drivers/lima/ir/pp/node.c

   1 /*
   2  * Copyright (c) 2017 Lima Project
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining a
   5  * copy of this software and associated documentation files (the "Software"),
   6  * to deal in the Software without restriction, including without limitation
   7  * the rights to use, copy, modify, merge, publish, distribute, sub license,
   8  * and/or sell copies of the Software, and to permit persons to whom the
   9  * Software is furnished to do so, subject to the following conditions:
  10  *
  11  * The above copyright notice and this permission notice (including the
  12  * next paragraph) shall be included in all copies or substantial portions
  13  * of the Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
  18  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  21  * DEALINGS IN THE SOFTWARE.
  22  *
  23  */
  24
  25 #include "util/u_math.h"
  26 #include "util/ralloc.h"
  27 #include "util/bitscan.h"
  28
  29 #include "ppir.h"
  30
  31 const ppir_op_info ppir_op_infos[] = {
  32    [ppir_op_mov] = {
  33       .name = "mov",
  34       .slots = (int []) {
  35          PPIR_INSTR_SLOT_ALU_SCL_ADD, PPIR_INSTR_SLOT_ALU_SCL_MUL,
  36          PPIR_INSTR_SLOT_ALU_VEC_ADD, PPIR_INSTR_SLOT_ALU_VEC_MUL,
  37          PPIR_INSTR_SLOT_END
  38       },
  39    },
  40    [ppir_op_abs] = {
  41       .name = "abs",
  42    },
  43    [ppir_op_neg] = {
  44       .name = "neg",
  45    },
  46    [ppir_op_sat] = {
  47       .name = "sat",
  48    },
  49    [ppir_op_mul] = {
  50       .name = "mul",
  51       .slots = (int []) {
  52          PPIR_INSTR_SLOT_ALU_SCL_MUL, PPIR_INSTR_SLOT_ALU_VEC_MUL,
  53          PPIR_INSTR_SLOT_END
  54       },
  55    },
  56    [ppir_op_add] = {
  57       .name = "add",
  58       .slots = (int []) {
  59          PPIR_INSTR_SLOT_ALU_SCL_ADD, PPIR_INSTR_SLOT_ALU_VEC_ADD,
  60          PPIR_INSTR_SLOT_END
  61       },
  62    },
  63    [ppir_op_sum3] = {
  64       .name = "sum3",
  65       .slots = (int []) {
  66          PPIR_INSTR_SLOT_ALU_VEC_ADD, PPIR_INSTR_SLOT_END
  67       },
  68    },
  69    [ppir_op_sum4] = {
  70       .name = "sum4",
  71       .slots = (int []) {
  72          PPIR_INSTR_SLOT_ALU_VEC_ADD, PPIR_INSTR_SLOT_END
  73       },
  74    },
  75    [ppir_op_rsqrt] = {
  76       .name = "rsqrt",
  77       .slots = (int []) {
  78          PPIR_INSTR_SLOT_ALU_COMBINE, PPIR_INSTR_SLOT_END
  79       },
  80    },
  81    [ppir_op_log2] = {
  82       .name = "log2",
  83       .slots = (int []) {
  84          PPIR_INSTR_SLOT_ALU_COMBINE, PPIR_INSTR_SLOT_END
  85       },
  86    },
  87    [ppir_op_exp2] = {
  88       .name = "exp2",
  89       .slots = (int []) {
  90          PPIR_INSTR_SLOT_ALU_COMBINE, PPIR_INSTR_SLOT_END
  91       },
  92    },
  93    [ppir_op_sqrt] = {
  94       .name = "sqrt",
  95       .slots = (int []) {
  96          PPIR_INSTR_SLOT_ALU_COMBINE, PPIR_INSTR_SLOT_END
  97       },
  98    },
  99    [ppir_op_sin] = {
 100       .name = "sin",
 101       .slots = (int []) {
 102          PPIR_INSTR_SLOT_ALU_COMBINE, PPIR_INSTR_SLOT_END
 103       },
 104    },
 105    [ppir_op_cos] = {
 106       .name = "cos",
 107       .slots = (int []) {
 108          PPIR_INSTR_SLOT_ALU_COMBINE, PPIR_INSTR_SLOT_END
 109       },
 110    },
 111    [ppir_op_max] = {
 112       .name = "max",
 113       .slots = (int []) {
 114          PPIR_INSTR_SLOT_ALU_SCL_ADD, PPIR_INSTR_SLOT_ALU_SCL_MUL,
 115          PPIR_INSTR_SLOT_ALU_VEC_ADD, PPIR_INSTR_SLOT_ALU_VEC_MUL,
 116          PPIR_INSTR_SLOT_END
 117       },
 118    },
 119    [ppir_op_min] = {
 120       .name = "min",
 121       .slots = (int []) {
 122          PPIR_INSTR_SLOT_ALU_SCL_ADD, PPIR_INSTR_SLOT_ALU_SCL_MUL,
 123          PPIR_INSTR_SLOT_ALU_VEC_ADD, PPIR_INSTR_SLOT_ALU_VEC_MUL,
 124          PPIR_INSTR_SLOT_END
 125       },
 126    },
 127    [ppir_op_floor] = {
 128       .name = "floor",
 129       .slots = (int []) {
 130          PPIR_INSTR_SLOT_ALU_SCL_ADD, PPIR_INSTR_SLOT_ALU_VEC_ADD,
 131          PPIR_INSTR_SLOT_END
 132       },
 133    },
 134    [ppir_op_ceil] = {
 135       .name = "ceil",
 136       .slots = (int []) {
 137          PPIR_INSTR_SLOT_ALU_SCL_ADD, PPIR_INSTR_SLOT_ALU_VEC_ADD,
 138          PPIR_INSTR_SLOT_END
 139       },
 140    },
 141    [ppir_op_fract] = {
 142       .name = "fract",
 143       .slots = (int []) {
 144          PPIR_INSTR_SLOT_ALU_SCL_ADD, PPIR_INSTR_SLOT_ALU_VEC_ADD,
 145          PPIR_INSTR_SLOT_END
 146       },
 147    },
 148    [ppir_op_ddx] = {
 149       .name = "ddx",
 150       .slots = (int []) {
 151          PPIR_INSTR_SLOT_ALU_SCL_ADD, PPIR_INSTR_SLOT_ALU_VEC_ADD,
 152          PPIR_INSTR_SLOT_END
 153       },
 154    },
 155    [ppir_op_ddy] = {
 156       .name = "ddy",
 157       .slots = (int []) {
 158          PPIR_INSTR_SLOT_ALU_SCL_ADD, PPIR_INSTR_SLOT_ALU_VEC_ADD,
 159          PPIR_INSTR_SLOT_END
 160       },
 161    },
 162    [ppir_op_and] = {
 163       .name = "and",
 164       .slots = (int []) {
 165          PPIR_INSTR_SLOT_ALU_SCL_MUL, PPIR_INSTR_SLOT_ALU_VEC_MUL,
 166          PPIR_INSTR_SLOT_END
 167       },
 168    },
 169    [ppir_op_or] = {
 170       .name = "or",
 171       .slots = (int []) {
 172          PPIR_INSTR_SLOT_ALU_SCL_MUL, PPIR_INSTR_SLOT_ALU_VEC_MUL,
 173          PPIR_INSTR_SLOT_END
 174       },
 175    },
 176    [ppir_op_xor] = {
 177       .name = "xor",
 178       .slots = (int []) {
 179          PPIR_INSTR_SLOT_ALU_SCL_MUL, PPIR_INSTR_SLOT_ALU_VEC_MUL,
 180          PPIR_INSTR_SLOT_END
 181       },
 182    },
 183    [ppir_op_not] = {
 184       .name = "not",
 185       .slots = (int []) {
 186          PPIR_INSTR_SLOT_ALU_SCL_MUL, PPIR_INSTR_SLOT_ALU_VEC_MUL,
 187          PPIR_INSTR_SLOT_END
 188       },
 189    },
 190    [ppir_op_lt] = {
 191       .name = "lt",
 192    },
 193    [ppir_op_le] = {
 194       .name = "le",
 195    },
 196    [ppir_op_gt] = {
 197       .name = "gt",
 198       .slots = (int []) {
 199          PPIR_INSTR_SLOT_ALU_SCL_MUL, PPIR_INSTR_SLOT_ALU_SCL_ADD,
 200          PPIR_INSTR_SLOT_ALU_VEC_MUL, PPIR_INSTR_SLOT_ALU_VEC_ADD,
 201          PPIR_INSTR_SLOT_END
 202       },
 203    },
 204    [ppir_op_ge] = {
 205       .name = "ge",
 206       .slots = (int []) {
 207          PPIR_INSTR_SLOT_ALU_SCL_MUL, PPIR_INSTR_SLOT_ALU_SCL_ADD,
 208          PPIR_INSTR_SLOT_ALU_VEC_MUL, PPIR_INSTR_SLOT_ALU_VEC_ADD,
 209          PPIR_INSTR_SLOT_END
 210       },
 211    },
 212    [ppir_op_eq] = {
 213       .name = "eq",
 214       .slots = (int []) {
 215          PPIR_INSTR_SLOT_ALU_SCL_MUL, PPIR_INSTR_SLOT_ALU_SCL_ADD,
 216          PPIR_INSTR_SLOT_ALU_VEC_MUL, PPIR_INSTR_SLOT_ALU_VEC_ADD,
 217          PPIR_INSTR_SLOT_END
 218       },
 219    },
 220    [ppir_op_ne] = {
 221       .name = "ne",
 222       .slots = (int []) {
 223          PPIR_INSTR_SLOT_ALU_SCL_MUL, PPIR_INSTR_SLOT_ALU_SCL_ADD,
 224          PPIR_INSTR_SLOT_ALU_VEC_MUL, PPIR_INSTR_SLOT_ALU_VEC_ADD,
 225          PPIR_INSTR_SLOT_END
 226       },
 227    },
 228    [ppir_op_sel_cond] = {
 229       /* effectively mov, but must be scheduled only to
 230        * PPIR_INSTR_SLOT_ALU_SCL_MUL */
 231       .name = "sel_cond",
 232       .slots = (int []) {
 233          PPIR_INSTR_SLOT_ALU_SCL_MUL, PPIR_INSTR_SLOT_END
 234       },
 235    },
 236    [ppir_op_select] = {
 237       .name = "select",
 238       .slots = (int []) {
 239          PPIR_INSTR_SLOT_ALU_SCL_ADD, PPIR_INSTR_SLOT_ALU_VEC_ADD,
 240          PPIR_INSTR_SLOT_END
 241       },
 242    },
 243    [ppir_op_rcp] = {
 244       .name = "rcp",
 245       .slots = (int []) {
 246          PPIR_INSTR_SLOT_ALU_COMBINE, PPIR_INSTR_SLOT_END
 247       },
 248    },
 249    [ppir_op_load_varying] = {
 250       .name = "ld_var",
 251       .type = ppir_node_type_load,
 252       .slots = (int []) {
 253          PPIR_INSTR_SLOT_VARYING, PPIR_INSTR_SLOT_END
 254       },
 255    },
 256    [ppir_op_load_coords] = {
 257       .name = "ld_coords",
 258       .type = ppir_node_type_load,
 259       .slots = (int []) {
 260          PPIR_INSTR_SLOT_VARYING, PPIR_INSTR_SLOT_END
 261       },
 262    },
 263    [ppir_op_load_fragcoord] = {
 264       .name = "ld_fragcoord",
 265       .type = ppir_node_type_load,
 266       .slots = (int []) {
 267          PPIR_INSTR_SLOT_VARYING, PPIR_INSTR_SLOT_END
 268       },
 269    },
 270    [ppir_op_load_pointcoord] = {
 271       .name = "ld_pointcoord",
 272       .type = ppir_node_type_load,
 273       .slots = (int []) {
 274          PPIR_INSTR_SLOT_VARYING, PPIR_INSTR_SLOT_END
 275       },
 276    },
 277    [ppir_op_load_frontface] = {
 278       .name = "ld_frontface",
 279       .type = ppir_node_type_load,
 280       .slots = (int []) {
 281          PPIR_INSTR_SLOT_VARYING, PPIR_INSTR_SLOT_END
 282       },
 283    },
 284    [ppir_op_load_uniform] = {
 285       .name = "ld_uni",
 286       .type = ppir_node_type_load,
 287       .slots = (int []) {
 288          PPIR_INSTR_SLOT_UNIFORM, PPIR_INSTR_SLOT_END
 289       },
 290    },
 291    [ppir_op_load_texture] = {
 292       .name = "ld_tex",
 293       .type = ppir_node_type_load_texture,
 294       .slots = (int []) {
 295          PPIR_INSTR_SLOT_TEXLD, PPIR_INSTR_SLOT_END
 296       },
 297    },
 298    [ppir_op_load_temp] = {
 299       .name = "ld_temp",
 300       .type = ppir_node_type_load,
 301       .slots = (int []) {
 302          PPIR_INSTR_SLOT_UNIFORM, PPIR_INSTR_SLOT_END
 303       },
 304    },
 305    [ppir_op_const] = {
 306       .name = "const",
 307       .type = ppir_node_type_const,
 308    },
 309    [ppir_op_store_color] = {
 310       .name = "st_col",
 311       .type = ppir_node_type_store,
 312    },
 313    [ppir_op_store_temp] = {
 314       .name = "st_temp",
 315       .type = ppir_node_type_store,
 316       .slots = (int []) {
 317          PPIR_INSTR_SLOT_STORE_TEMP, PPIR_INSTR_SLOT_END
 318       },
 319    },
 320    [ppir_op_discard] = {
 321       .name = "discard",
 322       .type = ppir_node_type_discard,
 323       .slots = (int []) {
 324          PPIR_INSTR_SLOT_BRANCH, PPIR_INSTR_SLOT_END
 325       },
 326    },
 327    [ppir_op_branch] = {
 328       .name = "branch",
 329       .type = ppir_node_type_branch,
 330       .slots = (int []) {
 331          PPIR_INSTR_SLOT_BRANCH, PPIR_INSTR_SLOT_END
 332       },
 333    },
 334 };
 335
 336 void *ppir_node_create(ppir_block *block, ppir_op op, int index, unsigned mask)
 337 {
 338    ppir_compiler *comp = block->comp;
 339    static const int node_size[] = {
 340       [ppir_node_type_alu] = sizeof(ppir_alu_node),
 341       [ppir_node_type_const] = sizeof(ppir_const_node),
 342       [ppir_node_type_load] = sizeof(ppir_load_node),
 343       [ppir_node_type_store] = sizeof(ppir_store_node),
 344       [ppir_node_type_load_texture] = sizeof(ppir_load_texture_node),
 345       [ppir_node_type_discard] = sizeof(ppir_discard_node),
 346       [ppir_node_type_branch] = sizeof(ppir_branch_node),
 347    };
 348
 349    ppir_node_type type = ppir_op_infos[op].type;
 350    int size = node_size[type];
 351    ppir_node *node = rzalloc_size(block, size);
 352    if (!node)
 353       return NULL;
 354
 355    list_inithead(&node->succ_list);
 356    list_inithead(&node->pred_list);
 357
 358    if (index >= 0) {
 359       if (mask) {
 360          /* reg has 4 slots for each componemt write node */
 361          while (mask)
 362             comp->var_nodes[(index << 2) + comp->reg_base + u_bit_scan(&mask)] = node;
 363          snprintf(node->name, sizeof(node->name), "reg%d", index);
 364       } else {
 365          comp->var_nodes[index] = node;
 366          snprintf(node->name, sizeof(node->name), "ssa%d", index);
 367       }
 368    }
 369    else
 370       snprintf(node->name, sizeof(node->name), "new");
 371
 372    node->op = op;
 373    node->type = type;
 374    node->index = comp->cur_index++;
 375    node->block = block;
 376
 377    return node;
 378 }
 379
 380 void ppir_node_add_dep(ppir_node *succ, ppir_node *pred)
 381 {
 382    /* don't add dep for two nodes from different block */
 383    if (succ->block != pred->block)
 384       return;
 385
 386    /* don't add duplicated dep */
 387    ppir_node_foreach_pred(succ, dep) {
 388       if (dep->pred == pred)
 389          return;
 390    }
 391
 392    ppir_dep *dep = ralloc(succ, ppir_dep);
 393    dep->pred = pred;
 394    dep->succ = succ;
 395    list_addtail(&dep->pred_link, &succ->pred_list);
 396    list_addtail(&dep->succ_link, &pred->succ_list);
 397 }
 398
 399 void ppir_node_remove_dep(ppir_dep *dep)
 400 {
 401    list_del(&dep->succ_link);
 402    list_del(&dep->pred_link);
 403    ralloc_free(dep);
 404 }
 405
 406 static void _ppir_node_replace_child(ppir_src *src, ppir_node *old_child, ppir_node *new_child)
 407 {
 408    ppir_dest *od = ppir_node_get_dest(old_child);
 409    if (ppir_node_target_equal(src, od)) {
 410       ppir_node_target_assign(src, new_child);
 411    }
 412 }
 413
 414 void ppir_node_replace_child(ppir_node *parent, ppir_node *old_child, ppir_node *new_child)
 415 {
 416    switch (parent->type) {
 417    case ppir_node_type_alu:
 418    {
 419       ppir_alu_node *alu = ppir_node_to_alu(parent);
 420       for (int i = 0; i < alu->num_src; i++)
 421          _ppir_node_replace_child(alu->src + i, old_child, new_child);
 422       break;
 423    }
 424    case ppir_node_type_branch:
 425    {
 426       ppir_branch_node *branch = ppir_node_to_branch(parent);
 427       for (int i = 0; i < 2; i++)
 428          _ppir_node_replace_child(branch->src + i, old_child, new_child);
 429       break;
 430    }
 431    case ppir_node_type_load:
 432    {
 433       ppir_load_node *load = ppir_node_to_load(parent);
 434       _ppir_node_replace_child(&load->src, old_child, new_child);
 435       break;
 436    }
 437    case ppir_node_type_load_texture:
 438    {
 439       ppir_load_texture_node *load_texture = ppir_node_to_load_texture(parent);
 440       _ppir_node_replace_child(&load_texture->src_coords, old_child, new_child);
 441       break;
 442    }
 443    case ppir_node_type_store:
 444    {
 445       ppir_store_node *store = ppir_node_to_store(parent);
 446       _ppir_node_replace_child(&store->src, old_child, new_child);
 447       break;
 448    }
 449    default:
 450       ppir_debug("unknown node type in %s\n", __func__);
 451       break;
 452    }
 453 }
 454
 455 void ppir_node_replace_pred(ppir_dep *dep, ppir_node *new_pred)
 456 {
 457    list_del(&dep->succ_link);
 458    dep->pred = new_pred;
 459    list_addtail(&dep->succ_link, &new_pred->succ_list);
 460 }
 461
 462 void ppir_node_replace_all_succ(ppir_node *dst, ppir_node *src)
 463 {
 464    ppir_node_foreach_succ_safe(src, dep) {
 465       ppir_node_replace_pred(dep, dst);
 466       ppir_node_replace_child(dep->succ, src, dst);
 467    }
 468 }
 469
 470 void ppir_node_delete(ppir_node *node)
 471 {
 472    ppir_node_foreach_succ_safe(node, dep)
 473       ppir_node_remove_dep(dep);
 474
 475    ppir_node_foreach_pred_safe(node, dep)
 476       ppir_node_remove_dep(dep);
 477
 478    list_del(&node->list);
 479    ralloc_free(node);
 480 }
 481
 482 static void ppir_node_print_dest(ppir_dest *dest)
 483 {
 484    switch (dest->type) {
 485    case ppir_target_ssa:
 486       printf("ssa%d", dest->ssa.index);
 487       break;
 488    case ppir_target_pipeline:
 489       printf("pipeline %d", dest->pipeline);
 490       break;
 491    case ppir_target_register:
 492       printf("reg %d", dest->reg->index);
 493       break;
 494    }
 495 }
 496
 497 static void ppir_node_print_src(ppir_src *src)
 498 {
 499    switch (src->type) {
 500    case ppir_target_ssa: {
 501       if (src->node)
 502          printf("ssa node %d", src->node->index);
 503       else
 504          printf("ssa idx %d", src->ssa ? src->ssa->index : -1);
 505       break;
 506    }
 507    case ppir_target_pipeline:
 508       if (src->node)
 509          printf("pipeline %d node %d", src->pipeline, src->node->index);
 510       else
 511          printf("pipeline %d", src->pipeline);
 512       break;
 513    case ppir_target_register:
 514       printf("reg %d", src->reg->index);
 515       break;
 516    }
 517 }
 518
 519 static void ppir_node_print_node(ppir_node *node, int space)
 520 {
 521    for (int i = 0; i < space; i++)
 522       printf(" ");
 523
 524    printf("%s%d: %s %s: ", node->printed && !ppir_node_is_leaf(node) ? "+" : "",
 525           node->index, ppir_op_infos[node->op].name, node->name);
 526
 527    ppir_dest *dest = ppir_node_get_dest(node);
 528    if (dest) {
 529       printf("dest: ");
 530       ppir_node_print_dest(dest);
 531    }
 532
 533    if (ppir_node_get_src_num(node) > 0) {
 534       printf(" src: ");
 535    }
 536    for (int i = 0; i < ppir_node_get_src_num(node); i++) {
 537       ppir_node_print_src(ppir_node_get_src(node, i));
 538       if (i != (ppir_node_get_src_num(node) - 1))
 539          printf(", ");
 540    }
 541    printf("\n");
 542
 543    if (!node->printed) {
 544       ppir_node_foreach_pred(node, dep) {
 545          ppir_node *pred = dep->pred;
 546          ppir_node_print_node(pred, space + 2);
 547       }
 548
 549       node->printed = true;
 550    }
 551 }
 552
 553 void ppir_node_print_prog(ppir_compiler *comp)
 554 {
 555    if (!(lima_debug & LIMA_DEBUG_PP))
 556       return;
 557
 558    list_for_each_entry(ppir_block, block, &comp->block_list, list) {
 559       list_for_each_entry(ppir_node, node, &block->node_list, list) {
 560          node->printed = false;
 561       }
 562    }
 563
 564    printf("========prog========\n");
 565    list_for_each_entry(ppir_block, block, &comp->block_list, list) {
 566       printf("-------block------\n");
 567       list_for_each_entry(ppir_node, node, &block->node_list, list) {
 568          if (ppir_node_is_root(node))
 569             ppir_node_print_node(node, 0);
 570       }
 571    }
 572    printf("====================\n");
 573 }
 574
 575 ppir_node *ppir_node_clone_const(ppir_block *block, ppir_node *node)
 576 {
 577    ppir_const_node *cnode = ppir_node_to_const(node);
 578    ppir_const_node *new_cnode = ppir_node_create(block, ppir_op_const, -1, 0);
 579
 580    if (!new_cnode)
 581       return NULL;
 582
 583    list_addtail(&new_cnode->node.list, &block->node_list);
 584
 585    new_cnode->constant.num = cnode->constant.num;
 586    for (int i = 0; i < cnode->constant.num; i++) {
 587       new_cnode->constant.value[i] = cnode->constant.value[i];
 588    }
 589    new_cnode->dest.type = ppir_target_ssa;
 590    new_cnode->dest.ssa.num_components = cnode->dest.ssa.num_components;
 591    new_cnode->dest.ssa.live_in = INT_MAX;
 592    new_cnode->dest.ssa.live_out = 0;
 593    new_cnode->dest.write_mask = cnode->dest.write_mask;
 594
 595    return &new_cnode->node;
 596 }