src/panfrost/bifrost/bir.c

   1 /*
   2  * Copyright (C) 2020 Collabora Ltd.
   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, sublicense,
   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 next
  12  * paragraph) shall be included in all copies or substantial portions of the
  13  * 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 NONINFRINGEMENT.  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 FROM,
  20  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  21  * SOFTWARE.
  22  *
  23  * Authors (Collabora):
  24  *      Alyssa Rosenzweig <alyssa.rosenzweig@collabora.com>
  25  */
  26
  27 #include "compiler.h"
  28
  29 /* Does an instruction respect outmods and source mods? Depend
  30  * on the types involved */
  31
  32 bool
  33 bi_has_outmod(bi_instruction *ins)
  34 {
  35         bool classy = bi_class_props[ins->type] & BI_MODS;
  36         bool floaty = nir_alu_type_get_base_type(ins->dest_type) == nir_type_float;
  37
  38         return classy && floaty;
  39 }
  40
  41 /* Have to check source for e.g. compares */
  42
  43 bool
  44 bi_has_source_mods(bi_instruction *ins)
  45 {
  46         bool classy = bi_class_props[ins->type] & BI_MODS;
  47         bool floaty = nir_alu_type_get_base_type(ins->src_types[0]) == nir_type_float;
  48
  49         return classy && floaty;
  50 }
  51
  52 /* A source is swizzled if the op is swizzlable, in 8-bit or
  53  * 16-bit mode, and the swizzled op. TODO: multi args */
  54
  55 bool
  56 bi_is_src_swizzled(bi_instruction *ins, unsigned s)
  57 {
  58         bool classy = bi_class_props[ins->type] & BI_SWIZZLABLE;
  59         bool small = nir_alu_type_get_type_size(ins->dest_type) < 32;
  60         bool first = (s == 0); /* TODO: prop? */
  61
  62         return classy && small && first;
  63 }
  64
  65 bool
  66 bi_has_arg(bi_instruction *ins, unsigned arg)
  67 {
  68         if (!ins)
  69                 return false;
  70
  71         bi_foreach_src(ins, s) {
  72                 if (ins->src[s] == arg)
  73                         return true;
  74         }
  75
  76         return false;
  77 }
  78
  79 uint16_t
  80 bi_from_bytemask(uint16_t bytemask, unsigned bytes)
  81 {
  82         unsigned value = 0;
  83
  84         for (unsigned c = 0, d = 0; c < 16; c += bytes, ++d) {
  85                 bool a = (bytemask & (1 << c)) != 0;
  86
  87                 for (unsigned q = c; q < bytes; ++q)
  88                         assert(((bytemask & (1 << q)) != 0) == a);
  89
  90                 value |= (a << d);
  91         }
  92
  93         return value;
  94 }
  95
  96 unsigned
  97 bi_get_component_count(bi_instruction *ins, signed src)
  98 {
  99         if (bi_class_props[ins->type] & BI_VECTOR) {
 100                 assert(ins->vector_channels);
 101                 return (src <= 0) ? ins->vector_channels : 1;
 102         } else {
 103                 unsigned bytes = nir_alu_type_get_type_size(src < 0 ? ins->dest_type : ins->src_types[src]);
 104
 105                 if (ins->type == BI_ATEST || ins->type == BI_SELECT)
 106                         return 1;
 107
 108                 return MAX2(32 / bytes, 1);
 109         }
 110 }
 111
 112 uint16_t
 113 bi_bytemask_of_read_components(bi_instruction *ins, unsigned node)
 114 {
 115         uint16_t mask = 0x0;
 116
 117         bi_foreach_src(ins, s) {
 118                 if (ins->src[s] != node) continue;
 119                 unsigned component_count = bi_get_component_count(ins, s);
 120                 nir_alu_type T = ins->src_types[s];
 121                 unsigned size = nir_alu_type_get_type_size(T);
 122                 unsigned bytes = size / 8;
 123                 unsigned cmask = (1 << bytes) - 1;
 124
 125                 for (unsigned i = 0; i < component_count; ++i) {
 126                         unsigned c = ins->swizzle[s][i];
 127                         mask |= (cmask << (c * bytes));
 128                 }
 129         }
 130
 131         return mask;
 132 }
 133
 134 uint64_t
 135 bi_get_immediate(bi_instruction *ins, unsigned index)
 136 {
 137         unsigned v = ins->src[index];
 138         assert(v & BIR_INDEX_CONSTANT);
 139         unsigned shift = v & ~BIR_INDEX_CONSTANT;
 140         uint64_t shifted = ins->constant.u64 >> shift;
 141
 142         /* Mask off the accessed part */
 143         unsigned sz = nir_alu_type_get_type_size(ins->src_types[index]);
 144
 145         if (sz == 64)
 146                 return shifted;
 147         else
 148                 return shifted & ((1ull << sz) - 1);
 149 }
 150
 151 bool
 152 bi_writes_component(bi_instruction *ins, unsigned comp)
 153 {
 154         return comp < bi_get_component_count(ins, -1);
 155 }
 156
 157 unsigned
 158 bi_writemask(bi_instruction *ins)
 159 {
 160         nir_alu_type T = ins->dest_type;
 161         unsigned size = nir_alu_type_get_type_size(T);
 162         unsigned bytes_per_comp = size / 8;
 163         unsigned components = bi_get_component_count(ins, -1);
 164         unsigned bytes = bytes_per_comp * components;
 165         unsigned mask = (1 << bytes) - 1;
 166         unsigned shift = ins->dest_offset * 4; /* 32-bit words */
 167         return (mask << shift);
 168 }