1/* 2 * Copyright © 2015 Intel Corporation 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 20 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS 21 * IN THE SOFTWARE. 22 * 23 * Authors: 24 * Jason Ekstrand <jason@jlekstrand.net> 25 */ 26 27#include "brw_nir.h" 28 29/* 30 * This file implements an analysis pass that determines when we have to do 31 * a boolean resolve on Gen <= 5. Instructions that need a boolean resolve 32 * will have the booleans portion of the instr->pass_flags field set to 33 * BRW_NIR_BOOLEAN_NEEDS_RESOLVE. 34 */ 35 36 37/** Returns the resolve status for the given source 38 * 39 * If the source has a parent instruction then the resolve status is the 40 * status of the parent instruction. If the source does not have a parent 41 * instruction then we don't know so we return NON_BOOLEAN. 42 */ 43static uint8_t 44get_resolve_status_for_src(nir_src *src) 45{ 46 if (src->is_ssa) { 47 nir_instr *src_instr = src->ssa->parent_instr; 48 uint8_t resolve_status = src_instr->pass_flags & BRW_NIR_BOOLEAN_MASK; 49 50 /* If the source instruction needs resolve, then from the perspective 51 * of the user, it's a true boolean. 52 */ 53 if (resolve_status == BRW_NIR_BOOLEAN_NEEDS_RESOLVE) 54 resolve_status = BRW_NIR_BOOLEAN_NO_RESOLVE; 55 return resolve_status; 56 } else { 57 return BRW_NIR_NON_BOOLEAN; 58 } 59} 60 61/** Marks the given source as needing a resolve 62 * 63 * If the given source corresponds to an unresolved boolean it marks it as 64 * needing a resolve. Otherwise, we leave it alone. 65 */ 66static bool 67src_mark_needs_resolve(nir_src *src, void *void_state) 68{ 69 if (src->is_ssa) { 70 nir_instr *src_instr = src->ssa->parent_instr; 71 uint8_t resolve_status = src_instr->pass_flags & BRW_NIR_BOOLEAN_MASK; 72 73 /* If the source instruction is unresolved, then mark it as needing 74 * to be resolved. 75 */ 76 if (resolve_status == BRW_NIR_BOOLEAN_UNRESOLVED) { 77 src_instr->pass_flags &= ~BRW_NIR_BOOLEAN_MASK; 78 src_instr->pass_flags |= BRW_NIR_BOOLEAN_NEEDS_RESOLVE; 79 } 80 81 } 82 83 return true; 84} 85 86static bool 87analyze_boolean_resolves_block(nir_block *block) 88{ 89 nir_foreach_instr(instr, block) { 90 switch (instr->type) { 91 case nir_instr_type_alu: { 92 /* For ALU instructions, the resolve status is handled in a 93 * three-step process. 94 * 95 * 1) Look at the instruction type and sources and determine if it 96 * can be left unresolved. 97 * 98 * 2) Look at the destination and see if we have to resolve 99 * anyway. (This is the case if this instruction is not the 100 * only instruction writing to a given register.) 101 * 102 * 3) If the instruction has a resolve status other than 103 * BOOL_UNRESOLVED or BOOL_NEEDS_RESOLVE then we walk through 104 * the sources and ensure that they are also resolved. This 105 * ensures that we don't end up with any stray unresolved 106 * booleans going into ADDs or something like that. 107 */ 108 109 uint8_t resolve_status; 110 nir_alu_instr *alu = nir_instr_as_alu(instr); 111 switch (alu->op) { 112 case nir_op_b32all_fequal2: 113 case nir_op_b32all_iequal2: 114 case nir_op_b32all_fequal3: 115 case nir_op_b32all_iequal3: 116 case nir_op_b32all_fequal4: 117 case nir_op_b32all_iequal4: 118 case nir_op_b32any_fnequal2: 119 case nir_op_b32any_inequal2: 120 case nir_op_b32any_fnequal3: 121 case nir_op_b32any_inequal3: 122 case nir_op_b32any_fnequal4: 123 case nir_op_b32any_inequal4: 124 /* These are only implemented by the vec4 backend and its 125 * implementation emits resolved booleans. At some point in the 126 * future, this may change and we'll have to remove some of the 127 * above cases. 128 */ 129 resolve_status = BRW_NIR_BOOLEAN_NO_RESOLVE; 130 break; 131 132 case nir_op_mov: 133 case nir_op_inot: 134 /* This is a single-source instruction. Just copy the resolve 135 * status from the source. 136 */ 137 resolve_status = get_resolve_status_for_src(&alu->src[0].src); 138 break; 139 140 case nir_op_b32csel: 141 case nir_op_iand: 142 case nir_op_ior: 143 case nir_op_ixor: { 144 const unsigned first = alu->op == nir_op_b32csel ? 1 : 0; 145 uint8_t src0_status = get_resolve_status_for_src(&alu->src[first + 0].src); 146 uint8_t src1_status = get_resolve_status_for_src(&alu->src[first + 1].src); 147 148 /* src0 of a bcsel is evaluated as a Boolean with the expectation 149 * that it has already been resolved. Mark it as such. 150 */ 151 if (alu->op == nir_op_b32csel) 152 src_mark_needs_resolve(&alu->src[0].src, NULL); 153 154 if (src0_status == src1_status) { 155 resolve_status = src0_status; 156 } else if (src0_status == BRW_NIR_NON_BOOLEAN || 157 src1_status == BRW_NIR_NON_BOOLEAN) { 158 /* If one of the sources is a non-boolean then the whole 159 * thing is a non-boolean. 160 */ 161 resolve_status = BRW_NIR_NON_BOOLEAN; 162 } else { 163 /* At this point one of them is a true boolean and one is a 164 * boolean that needs a resolve. We could either resolve the 165 * unresolved source or we could resolve here. If we resolve 166 * the unresolved source then we get two resolves for the price 167 * of one. Just set this one to BOOLEAN_NO_RESOLVE and we'll 168 * let the code below force a resolve on the unresolved source. 169 */ 170 resolve_status = BRW_NIR_BOOLEAN_NO_RESOLVE; 171 } 172 break; 173 } 174 175 default: 176 if (nir_alu_type_get_base_type(nir_op_infos[alu->op].output_type) == nir_type_bool) { 177 /* This instructions will turn into a CMP when we actually emit 178 * them so the result will have to be resolved before it can be 179 * used. 180 */ 181 resolve_status = BRW_NIR_BOOLEAN_UNRESOLVED; 182 183 /* Even though the destination is allowed to be left 184 * unresolved, the sources are treated as regular integers or 185 * floats so they need to be resolved. 186 */ 187 nir_foreach_src(instr, src_mark_needs_resolve, NULL); 188 } else { 189 resolve_status = BRW_NIR_NON_BOOLEAN; 190 } 191 } 192 193 /* If the destination is SSA, go ahead allow unresolved booleans. 194 * If the destination register doesn't have a well-defined parent_instr 195 * we need to resolve immediately. 196 */ 197 if (!alu->dest.dest.is_ssa && 198 resolve_status == BRW_NIR_BOOLEAN_UNRESOLVED) { 199 resolve_status = BRW_NIR_BOOLEAN_NEEDS_RESOLVE; 200 } 201 202 instr->pass_flags = (instr->pass_flags & ~BRW_NIR_BOOLEAN_MASK) | 203 resolve_status; 204 205 /* Finally, resolve sources if it's needed */ 206 switch (resolve_status) { 207 case BRW_NIR_BOOLEAN_NEEDS_RESOLVE: 208 case BRW_NIR_BOOLEAN_UNRESOLVED: 209 /* This instruction is either unresolved or we're doing the 210 * resolve here; leave the sources alone. 211 */ 212 break; 213 214 case BRW_NIR_BOOLEAN_NO_RESOLVE: 215 case BRW_NIR_NON_BOOLEAN: 216 nir_foreach_src(instr, src_mark_needs_resolve, NULL); 217 break; 218 219 default: 220 unreachable("Invalid boolean flag"); 221 } 222 223 break; 224 } 225 226 case nir_instr_type_load_const: { 227 nir_load_const_instr *load = nir_instr_as_load_const(instr); 228 229 /* For load_const instructions, it's a boolean exactly when it holds 230 * one of the values NIR_TRUE or NIR_FALSE. 231 * 232 * Since load_const instructions don't have any sources, we don't 233 * have to worry about resolving them. 234 */ 235 instr->pass_flags &= ~BRW_NIR_BOOLEAN_MASK; 236 if (load->value[0].u32 == NIR_TRUE || load->value[0].u32 == NIR_FALSE) { 237 instr->pass_flags |= BRW_NIR_BOOLEAN_NO_RESOLVE; 238 } else { 239 instr->pass_flags |= BRW_NIR_NON_BOOLEAN; 240 } 241 continue; 242 } 243 244 default: 245 /* Everything else is an unknown non-boolean value and needs to 246 * have all sources resolved. 247 */ 248 instr->pass_flags = (instr->pass_flags & ~BRW_NIR_BOOLEAN_MASK) | 249 BRW_NIR_NON_BOOLEAN; 250 nir_foreach_src(instr, src_mark_needs_resolve, NULL); 251 continue; 252 } 253 } 254 255 nir_if *following_if = nir_block_get_following_if(block); 256 if (following_if) 257 src_mark_needs_resolve(&following_if->condition, NULL); 258 259 return true; 260} 261 262static void 263analyze_boolean_resolves_impl(nir_function_impl *impl) 264{ 265 nir_foreach_block(block, impl) { 266 analyze_boolean_resolves_block(block); 267 } 268} 269 270void 271brw_nir_analyze_boolean_resolves(nir_shader *shader) 272{ 273 nir_foreach_function(function, shader) { 274 if (function->impl) 275 analyze_boolean_resolves_impl(function->impl); 276 } 277} 278