src/compiler/nir/nir_builder_opcodes_h.py

   1 from __future__ import print_function
   2
   3 template = """\
   4 /* Copyright (C) 2015 Broadcom
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the "Software"),
   8  * to deal in the Software without restriction, including without limitation
   9  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  10  * and/or sell copies of the Software, and to permit persons to whom the
  11  * Software is furnished to do so, subject to the following conditions:
  12  *
  13  * The above copyright notice and this permission notice (including the next
  14  * paragraph) shall be included in all copies or substantial portions of the
  15  * Software.
  16  *
  17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  18  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  19  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  20  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  21  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  22  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  23  * IN THE SOFTWARE.
  24  */
  25
  26 #ifndef _NIR_BUILDER_OPCODES_
  27 #define _NIR_BUILDER_OPCODES_
  28
  29 <%
  30 def src_decl_list(num_srcs):
  31    return ', '.join('nir_ssa_def *src' + str(i) for i in range(num_srcs))
  32
  33 def src_list(num_srcs):
  34    if num_srcs <= 4:
  35       return ', '.join('src' + str(i) if i < num_srcs else 'NULL' for i in range(4))
  36    else:
  37       return ', '.join('src' + str(i) for i in range(num_srcs))
  38 %>
  39
  40 % for name, opcode in sorted(opcodes.items()):
  41 static inline nir_ssa_def *
  42 nir_${name}(nir_builder *build, ${src_decl_list(opcode.num_inputs)})
  43 {
  44 % if opcode.num_inputs <= 4:
  45    return nir_build_alu(build, nir_op_${name}, ${src_list(opcode.num_inputs)});
  46 % else:
  47    nir_ssa_def *srcs[${opcode.num_inputs}] = {${src_list(opcode.num_inputs)}};
  48    return nir_build_alu_src_arr(build, nir_op_${name}, srcs);
  49 % endif
  50 }
  51 % endfor
  52
  53 /* Generic builder for system values. */
  54 static inline nir_ssa_def *
  55 nir_load_system_value(nir_builder *build, nir_intrinsic_op op, int index,
  56                       unsigned num_components, unsigned bit_size)
  57 {
  58    nir_intrinsic_instr *load = nir_intrinsic_instr_create(build->shader, op);
  59    if (nir_intrinsic_infos[op].dest_components > 0)
  60       assert(num_components == nir_intrinsic_infos[op].dest_components);
  61    else
  62       load->num_components = num_components;
  63    load->const_index[0] = index;
  64
  65    nir_ssa_dest_init(&load->instr, &load->dest,
  66                      num_components, bit_size, NULL);
  67    nir_builder_instr_insert(build, &load->instr);
  68    return &load->dest.ssa;
  69 }
  70
  71 <%
  72 def sysval_decl_list(opcode):
  73    res = ''
  74    if opcode.indices:
  75       res += ', unsigned ' + opcode.indices[0].lower()
  76    if opcode.dest_components == 0:
  77       res += ', unsigned num_components'
  78    if len(opcode.bit_sizes) != 1:
  79       res += ', unsigned bit_size'
  80    return res
  81
  82 def sysval_arg_list(opcode):
  83    args = []
  84    if opcode.indices:
  85       args.append(opcode.indices[0].lower())
  86    else:
  87       args.append('0')
  88
  89    if opcode.dest_components == 0:
  90       args.append('num_components')
  91    else:
  92       args.append(str(opcode.dest_components))
  93
  94    if len(opcode.bit_sizes) == 1:
  95       bit_size = opcode.bit_sizes[0]
  96       args.append(str(bit_size))
  97    else:
  98       args.append('bit_size')
  99
 100    return ', '.join(args)
 101 %>
 102
 103 % for name, opcode in filter(lambda v: v[1].sysval, sorted(INTR_OPCODES.items())):
 104 <% assert len(opcode.bit_sizes) > 0 %>
 105 static inline nir_ssa_def *
 106 nir_${name}(nir_builder *build${sysval_decl_list(opcode)})
 107 {
 108    return nir_load_system_value(build, nir_intrinsic_${name},
 109                                 ${sysval_arg_list(opcode)});
 110 }
 111 % endfor
 112
 113 #endif /* _NIR_BUILDER_OPCODES_ */"""
 114
 115 from nir_opcodes import opcodes
 116 from nir_intrinsics import INTR_OPCODES
 117 from mako.template import Template
 118
 119 print(Template(template).render(opcodes=opcodes, INTR_OPCODES=INTR_OPCODES))