src/gallium/drivers/lima/ir/pp/node_to_instr.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 "ppir.h"
  26
  27
  28 static bool create_new_instr(ppir_block *block, ppir_node *node)
  29 {
  30    ppir_instr *instr = ppir_instr_create(block);
  31    if (unlikely(!instr))
  32       return false;
  33
  34    if (!ppir_instr_insert_node(instr, node))
  35       return false;
  36
  37    return true;
  38 }
  39
  40 static bool insert_to_load_tex(ppir_block *block, ppir_node *load_coords, ppir_node *ldtex)
  41 {
  42    ppir_dest *dest = ppir_node_get_dest(ldtex);
  43    ppir_node *move = NULL;
  44
  45    ppir_load_node *load = ppir_node_to_load(load_coords);
  46    load->dest.type = ppir_target_pipeline;
  47    load->dest.pipeline = ppir_pipeline_reg_discard;
  48
  49    ppir_load_texture_node *load_texture = ppir_node_to_load_texture(ldtex);
  50    load_texture->src_coords.type = ppir_target_pipeline;
  51    load_texture->src_coords.pipeline = ppir_pipeline_reg_discard;
  52
  53    /* Insert load_coords to ldtex instruction */
  54    if (!ppir_instr_insert_node(ldtex->instr, load_coords))
  55       return false;
  56
  57    /* Create move node */
  58    move = ppir_node_create(block, ppir_op_mov, -1 , 0);
  59    if (unlikely(!move))
  60       return false;
  61
  62    ppir_debug("insert_load_tex: create move %d for %d\n",
  63               move->index, ldtex->index);
  64
  65    ppir_alu_node *alu = ppir_node_to_alu(move);
  66    alu->dest = *dest;
  67
  68    ppir_node_replace_all_succ(move, ldtex);
  69
  70    dest->type = ppir_target_pipeline;
  71    dest->pipeline = ppir_pipeline_reg_sampler;
  72
  73    alu->num_src = 1;
  74    ppir_node_target_assign(&alu->src[0], dest);
  75    for (int i = 0; i < 4; i++)
  76       alu->src->swizzle[i] = i;
  77
  78    ppir_node_add_dep(move, ldtex);
  79    list_addtail(&move->list, &ldtex->list);
  80
  81    if (!ppir_instr_insert_node(ldtex->instr, move))
  82       return false;
  83
  84    return true;
  85 }
  86
  87 static bool insert_to_each_succ_instr(ppir_block *block, ppir_node *node)
  88 {
  89    ppir_dest *dest = ppir_node_get_dest(node);
  90    assert(dest->type == ppir_target_ssa);
  91
  92    ppir_node *move = NULL;
  93
  94    ppir_node_foreach_succ_safe(node, dep) {
  95       ppir_node *succ = dep->succ;
  96       assert(succ->type == ppir_node_type_alu ||
  97              succ->type == ppir_node_type_branch);
  98
  99       if (!ppir_instr_insert_node(succ->instr, node)) {
 100          /* create a move node to insert for failed node */
 101          if (!move) {
 102             move = ppir_node_create(block, ppir_op_mov, -1, 0);
 103             if (unlikely(!move))
 104                return false;
 105
 106             ppir_debug("node_to_instr create move %d for %d\n",
 107                        move->index, node->index);
 108
 109             ppir_alu_node *alu = ppir_node_to_alu(move);
 110             alu->dest = *dest;
 111             alu->num_src = 1;
 112             ppir_node_target_assign(alu->src, dest);
 113             for (int i = 0; i < 4; i++)
 114                alu->src->swizzle[i] = i;
 115          }
 116
 117          ppir_node_replace_pred(dep, move);
 118          ppir_node_replace_child(succ, node, move);
 119       }
 120    }
 121
 122    if (move) {
 123       if (!create_new_instr(block, move))
 124          return false;
 125
 126       MAYBE_UNUSED bool insert_result =
 127          ppir_instr_insert_node(move->instr, node);
 128       assert(insert_result);
 129
 130       ppir_node_add_dep(move, node);
 131       list_addtail(&move->list, &node->list);
 132    }
 133
 134    /* dupliacte node for each successor */
 135
 136    bool first = true;
 137    struct list_head dup_list;
 138    list_inithead(&dup_list);
 139
 140    ppir_node_foreach_succ_safe(node, dep) {
 141       ppir_node *succ = dep->succ;
 142
 143       if (first) {
 144          first = false;
 145          node->instr = succ->instr;
 146          continue;
 147       }
 148
 149       if (succ->instr == node->instr)
 150          continue;
 151
 152       list_for_each_entry(ppir_node, dup, &dup_list, list) {
 153          if (succ->instr == dup->instr) {
 154             ppir_node_replace_pred(dep, dup);
 155             continue;
 156          }
 157       }
 158
 159       ppir_node *dup = ppir_node_create(block, node->op, -1, 0);
 160       if (unlikely(!dup))
 161          return false;
 162       list_addtail(&dup->list, &dup_list);
 163
 164       ppir_debug("node_to_instr duplicate %s %d from %d\n",
 165                  ppir_op_infos[dup->op].name, dup->index, node->index);
 166
 167       ppir_instr *instr = succ->instr;
 168       dup->instr = instr;
 169       dup->instr_pos = node->instr_pos;
 170       ppir_node_replace_pred(dep, dup);
 171
 172       if ((node->op == ppir_op_load_uniform) || (node->op == ppir_op_load_temp)) {
 173          ppir_load_node *load = ppir_node_to_load(node);
 174          ppir_load_node *dup_load = ppir_node_to_load(dup);
 175          dup_load->dest = load->dest;
 176          dup_load->index = load->index;
 177          dup_load->num_components = load->num_components;
 178          instr->slots[node->instr_pos] = dup;
 179       }
 180    }
 181
 182    list_splicetail(&dup_list, &node->list);
 183
 184    return true;
 185 }
 186
 187 static bool ppir_do_node_to_instr(ppir_block *block, ppir_node *node)
 188 {
 189    switch (node->type) {
 190    case ppir_node_type_alu:
 191    {
 192       /* merge pred mul and succ add in the same instr can save a reg
 193        * by using pipeline reg ^vmul/^fmul */
 194       ppir_alu_node *alu = ppir_node_to_alu(node);
 195       if (alu->dest.type == ppir_target_ssa &&
 196           ppir_node_has_single_succ(node)) {
 197          ppir_node *succ = ppir_node_first_succ(node);
 198          if (succ->instr_pos == PPIR_INSTR_SLOT_ALU_VEC_ADD) {
 199             node->instr_pos = PPIR_INSTR_SLOT_ALU_VEC_MUL;
 200             /* select instr's condition must be inserted to fmul slot */
 201             if (succ->op == ppir_op_select &&
 202                 ppir_node_first_pred(succ) == node) {
 203                assert(alu->dest.ssa.num_components == 1);
 204                node->instr_pos = PPIR_INSTR_SLOT_ALU_SCL_MUL;
 205             }
 206             ppir_instr_insert_mul_node(succ, node);
 207          }
 208          else if (succ->instr_pos == PPIR_INSTR_SLOT_ALU_SCL_ADD &&
 209                   alu->dest.ssa.num_components == 1) {
 210             node->instr_pos = PPIR_INSTR_SLOT_ALU_SCL_MUL;
 211             ppir_instr_insert_mul_node(succ, node);
 212          }
 213       }
 214
 215       /* can't inserted to any existing instr, create one */
 216       if (!node->instr && !create_new_instr(block, node))
 217          return false;
 218
 219       break;
 220    }
 221    case ppir_node_type_load:
 222       if ((node->op == ppir_op_load_uniform) || (node->op == ppir_op_load_temp)) {
 223          /* merge pred load_uniform into succ instr can save a reg
 224           * by using pipeline reg */
 225          if (!insert_to_each_succ_instr(block, node))
 226             return false;
 227
 228          ppir_load_node *load = ppir_node_to_load(node);
 229          load->dest.type = ppir_target_pipeline;
 230          load->dest.pipeline = ppir_pipeline_reg_uniform;
 231       }
 232       else if (node->op == ppir_op_load_temp) {
 233          /* merge pred load_temp into succ instr can save a reg
 234           * by using pipeline reg */
 235          if (!insert_to_each_succ_instr(block, node))
 236             return false;
 237
 238          ppir_load_node *load = ppir_node_to_load(node);
 239          load->dest.type = ppir_target_pipeline;
 240          load->dest.pipeline = ppir_pipeline_reg_uniform;
 241       }
 242       else if (node->op == ppir_op_load_varying ||
 243                node->op == ppir_op_load_fragcoord) {
 244          /* delay the load varying dup to scheduler */
 245          if (!create_new_instr(block, node))
 246             return false;
 247       }
 248       else if (node->op == ppir_op_load_coords) {
 249          ppir_node *ldtex = ppir_node_first_succ(node);
 250          if (!insert_to_load_tex(block, node, ldtex))
 251             return false;
 252       }
 253       else {
 254          /* not supported yet */
 255          assert(0);
 256          return false;
 257       }
 258       break;
 259    case ppir_node_type_load_texture:
 260       if (!create_new_instr(block, node))
 261          return false;
 262       break;
 263    case ppir_node_type_const:
 264       if (!insert_to_each_succ_instr(block, node))
 265          return false;
 266       break;
 267    case ppir_node_type_store:
 268    {
 269       if (node->op == ppir_op_store_temp) {
 270          if (!create_new_instr(block, node))
 271             return false;
 272          break;
 273       }
 274
 275       /* Only the store color node should appear here.
 276        * Currently we always insert a move node as the end instr.
 277        * But it should only be done when:
 278        *   1. store a const node
 279        *   2. store a load node
 280        *   3. store a reg assigned in another block like loop/if
 281        */
 282
 283       assert(node->op == ppir_op_store_color);
 284
 285       ppir_node *move = ppir_node_create(block, ppir_op_mov, -1, 0);
 286       if (unlikely(!move))
 287          return false;
 288
 289       ppir_debug("node_to_instr create move %d from store %d\n",
 290                  move->index, node->index);
 291
 292       ppir_node_foreach_pred_safe(node, dep) {
 293          ppir_node *pred = dep->pred;
 294          /* we can't do this in this function except here as this
 295           * store is the root of this recursion */
 296          ppir_node_remove_dep(dep);
 297          ppir_node_add_dep(move, pred);
 298       }
 299
 300       ppir_node_add_dep(node, move);
 301       list_addtail(&move->list, &node->list);
 302
 303       ppir_alu_node *alu = ppir_node_to_alu(move);
 304       ppir_store_node *store = ppir_node_to_store(node);
 305       alu->src[0] = store->src;
 306       alu->num_src = 1;
 307
 308       alu->dest.type = ppir_target_ssa;
 309       alu->dest.ssa.num_components = 4;
 310       alu->dest.ssa.live_in = INT_MAX;
 311       alu->dest.ssa.live_out = 0;
 312       alu->dest.write_mask = 0xf;
 313
 314       store->src.type = ppir_target_ssa;
 315       store->src.ssa = &alu->dest.ssa;
 316
 317       if (!create_new_instr(block, move))
 318          return false;
 319
 320       move->instr->is_end = true;
 321       node->instr = move->instr;
 322
 323       /* use move for the following recursion */
 324       node = move;
 325       break;
 326    }
 327    case ppir_node_type_discard:
 328       if (!create_new_instr(block, node))
 329          return false;
 330       node->instr->is_end = true;
 331       break;
 332    case ppir_node_type_branch:
 333       if (!create_new_instr(block, node))
 334          return false;
 335       break;
 336    default:
 337       return false;
 338    }
 339
 340    /* we have to make sure the dep not be destroyed (due to
 341     * succ change) in ppir_do_node_to_instr, otherwise we can't
 342     * do recursion like this */
 343    ppir_node_foreach_pred(node, dep) {
 344       ppir_node *pred = dep->pred;
 345       bool ready = true;
 346
 347       /* pred may already be processed by the previous pred
 348        * (this pred may be both node and previous pred's child) */
 349       if (pred->instr)
 350          continue;
 351
 352       /* insert pred only when all its successors have been inserted */
 353       ppir_node_foreach_succ(pred, dep) {
 354          ppir_node *succ = dep->succ;
 355          if (!succ->instr) {
 356             ready = false;
 357             break;
 358          }
 359       }
 360
 361       if (ready) {
 362          if (!ppir_do_node_to_instr(block, pred))
 363             return false;
 364       }
 365    }
 366
 367    return true;
 368 }
 369
 370 static bool ppir_create_instr_from_node(ppir_compiler *comp)
 371 {
 372    list_for_each_entry(ppir_block, block, &comp->block_list, list) {
 373       list_for_each_entry(ppir_node, node, &block->node_list, list) {
 374          if (ppir_node_is_root(node)) {
 375             if (!ppir_do_node_to_instr(block, node))
 376                return false;
 377          }
 378       }
 379    }
 380
 381    return true;
 382 }
 383
 384 static void ppir_build_instr_dependency(ppir_compiler *comp)
 385 {
 386    list_for_each_entry(ppir_block, block, &comp->block_list, list) {
 387       list_for_each_entry(ppir_instr, instr, &block->instr_list, list) {
 388          for (int i = 0; i < PPIR_INSTR_SLOT_NUM; i++) {
 389             ppir_node *node = instr->slots[i];
 390             if (node) {
 391                ppir_node_foreach_pred(node, dep) {
 392                   ppir_node *pred = dep->pred;
 393                   if (pred->instr && pred->instr != instr)
 394                      ppir_instr_add_dep(instr, pred->instr);
 395                }
 396             }
 397          }
 398       }
 399    }
 400 }
 401
 402 bool ppir_node_to_instr(ppir_compiler *comp)
 403 {
 404    if (!ppir_create_instr_from_node(comp))
 405       return false;
 406    ppir_instr_print_list(comp);
 407
 408    ppir_build_instr_dependency(comp);
 409    ppir_instr_print_dep(comp);
 410
 411    return true;
 412 }