1aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand/* 2aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * Copyright © 2014 Intel Corporation 3aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * 4aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * Permission is hereby granted, free of charge, to any person obtaining a 5aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * copy of this software and associated documentation files (the "Software"), 6aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * to deal in the Software without restriction, including without limitation 7aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * the rights to use, copy, modify, merge, publish, distribute, sublicense, 8aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * and/or sell copies of the Software, and to permit persons to whom the 9aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * Software is furnished to do so, subject to the following conditions: 10aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * 11aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * The above copyright notice and this permission notice (including the next 12aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * paragraph) shall be included in all copies or substantial portions of the 13aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * Software. 14aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * 15aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 16aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 17aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL 18aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 19aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING 20aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS 21aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * IN THE SOFTWARE. 22aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * 23aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * Authors: 24aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * Jason Ekstrand (jason@jlekstrand.net) 25aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * 26aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand */ 27aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 28aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand#include "nir.h" 2906f3c98b9da35b5f5c02bd30599fbde57a19520eJason Ekstrand#include "nir_array.h" 30aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 31aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrandstruct locals_to_regs_state { 324e9b3765947f2c98d618082cd449db2e319bd887Jason Ekstrand nir_shader *shader; 33aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand nir_function_impl *impl; 34aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 35aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand /* A hash table mapping derefs to registers */ 36aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand struct hash_table *regs_table; 3706f3c98b9da35b5f5c02bd30599fbde57a19520eJason Ekstrand 38cfae0f8a3ae98ae10bcf9e4f1429f66165a17d95Kenneth Graunke bool progress; 39aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand}; 40aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 41aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand/* The following two functions implement a hash and equality check for 42aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * variable dreferences. When the hash or equality function encounters an 43aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * array, it ignores the offset and whether it is direct or indirect 44aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand * entirely. 45aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand */ 46aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrandstatic uint32_t 47aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrandhash_deref(const void *void_deref) 48aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand{ 490c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand uint32_t hash = _mesa_fnv32_1a_offset_bias; 50aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 510c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand const nir_deref_var *deref_var = void_deref; 520c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand hash = _mesa_fnv32_1a_accumulate(hash, deref_var->var); 53aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 540c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand for (const nir_deref *deref = deref_var->deref.child; 550c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand deref; deref = deref->child) { 560c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand if (deref->deref_type == nir_deref_type_struct) { 570c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand const nir_deref_struct *deref_struct = nir_deref_as_struct(deref); 580c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand hash = _mesa_fnv32_1a_accumulate(hash, deref_struct->index); 590c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand } 60aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand } 61aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 620c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand return hash; 63aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand} 64aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 65aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrandstatic bool 66aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrandderefs_equal(const void *void_a, const void *void_b) 67aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand{ 680c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand const nir_deref_var *a_var = void_a; 690c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand const nir_deref_var *b_var = void_b; 70aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 710c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand if (a_var->var != b_var->var) 72aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand return false; 73aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 740c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand for (const nir_deref *a = a_var->deref.child, *b = b_var->deref.child; 750c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand a != NULL; a = a->child, b = b->child) { 760c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand if (a->deref_type != b->deref_type) 77aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand return false; 780c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand 790c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand if (a->deref_type == nir_deref_type_struct) { 800c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand if (nir_deref_as_struct(a)->index != nir_deref_as_struct(b)->index) 810c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand return false; 820c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand } 830c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand /* Do nothing for arrays. They're all the same. */ 840c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand 850c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand assert((a->child == NULL) == (b->child == NULL)); 860c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand if((a->child == NULL) != (b->child == NULL)) 87aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand return false; 88aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand } 89aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 900c0ca8b6aec0842e396b81ef984ad01e00883c3eJason Ekstrand return true; 91aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand} 92aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 93aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrandstatic nir_register * 94aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrandget_reg_for_deref(nir_deref_var *deref, struct locals_to_regs_state *state) 95aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand{ 96aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand uint32_t hash = hash_deref(deref); 97aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 9850e0b0bee3fb97089cf9913af7a0c980dfce6dceJason Ekstrand assert(deref->var->constant_initializer == NULL); 9950e0b0bee3fb97089cf9913af7a0c980dfce6dceJason Ekstrand 100aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand struct hash_entry *entry = 101aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand _mesa_hash_table_search_pre_hashed(state->regs_table, hash, deref); 102aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand if (entry) 103aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand return entry->data; 104aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 105aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand unsigned array_size = 1; 106aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand nir_deref *tail = &deref->deref; 107aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand while (tail->child) { 1088b900e74058dd48368511780a488ccb7a645c64fJason Ekstrand if (tail->child->deref_type == nir_deref_type_array) 1098b900e74058dd48368511780a488ccb7a645c64fJason Ekstrand array_size *= glsl_get_length(tail->type); 110aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand tail = tail->child; 111aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand } 112aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 113aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand assert(glsl_type_is_vector(tail->type) || glsl_type_is_scalar(tail->type)); 114aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 115aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand nir_register *reg = nir_local_reg_create(state->impl); 116aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand reg->num_components = glsl_get_vector_elements(tail->type); 117aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand reg->num_array_elems = array_size > 1 ? array_size : 0; 1189d3cc80b752b236bc51b78a3d99920748a1a230aRob Clark reg->bit_size = glsl_get_bit_size(tail->type); 119aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 1208ed5305d28d9309d651dfec3fbf4349854694694Jason Ekstrand _mesa_hash_table_insert_pre_hashed(state->regs_table, hash, deref, reg); 121aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 122aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand return reg; 123aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand} 124aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 125aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrandstatic nir_src 126aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrandget_deref_reg_src(nir_deref_var *deref, nir_instr *instr, 127aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand struct locals_to_regs_state *state) 128aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand{ 129aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand nir_src src; 130aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 131aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand src.is_ssa = false; 132aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand src.reg.reg = get_reg_for_deref(deref, state); 133aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand src.reg.base_offset = 0; 134aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand src.reg.indirect = NULL; 135aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 136d61bd972d861d9246fe7f9de71158aa79368bb79Jason Ekstrand /* It is possible for a user to create a shader that has an array with a 137d61bd972d861d9246fe7f9de71158aa79368bb79Jason Ekstrand * single element and then proceed to access it indirectly. Indirectly 138d61bd972d861d9246fe7f9de71158aa79368bb79Jason Ekstrand * accessing a non-array register is not allowed in NIR. In order to 139d61bd972d861d9246fe7f9de71158aa79368bb79Jason Ekstrand * handle this case we just convert it to a direct reference. 140d61bd972d861d9246fe7f9de71158aa79368bb79Jason Ekstrand */ 141d61bd972d861d9246fe7f9de71158aa79368bb79Jason Ekstrand if (src.reg.reg->num_array_elems == 0) 142d61bd972d861d9246fe7f9de71158aa79368bb79Jason Ekstrand return src; 143d61bd972d861d9246fe7f9de71158aa79368bb79Jason Ekstrand 144aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand nir_deref *tail = &deref->deref; 145aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand while (tail->child != NULL) { 146aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand const struct glsl_type *parent_type = tail->type; 147aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand tail = tail->child; 148aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 149aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand if (tail->deref_type != nir_deref_type_array) 150aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand continue; 151aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 152aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand nir_deref_array *deref_array = nir_deref_as_array(tail); 153aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 154aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand src.reg.base_offset *= glsl_get_length(parent_type); 155aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand src.reg.base_offset += deref_array->base_offset; 156aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 157aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand if (src.reg.indirect) { 158aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand nir_load_const_instr *load_const = 1593663a2397e47da9b766b0c4239a8b74ac77b5d04Samuel Iglesias Gonsálvez nir_load_const_instr_create(state->shader, 1, 32); 160084b24f5582567ebf5aa94b7f40ae3bdcb71316bIago Toral Quiroga load_const->value.u32[0] = glsl_get_length(parent_type); 161aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand nir_instr_insert_before(instr, &load_const->instr); 162aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 1634e9b3765947f2c98d618082cd449db2e319bd887Jason Ekstrand nir_alu_instr *mul = nir_alu_instr_create(state->shader, nir_op_imul); 164aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand mul->src[0].src = *src.reg.indirect; 165aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand mul->src[1].src.is_ssa = true; 1662c7da78805175f36879111306ac37c12d33bf65bJason Ekstrand mul->src[1].src.ssa = &load_const->def; 167aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand mul->dest.write_mask = 1; 1683124ce699bb3844e793f00e00bfbea5c91744f90Connor Abbott nir_ssa_dest_init(&mul->instr, &mul->dest.dest, 1, 32, NULL); 169aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand nir_instr_insert_before(instr, &mul->instr); 170aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 171aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand src.reg.indirect->is_ssa = true; 172aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand src.reg.indirect->ssa = &mul->dest.dest.ssa; 173aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand } 174aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 175aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand if (deref_array->deref_array_type == nir_deref_array_type_indirect) { 176aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand if (src.reg.indirect == NULL) { 1774e9b3765947f2c98d618082cd449db2e319bd887Jason Ekstrand src.reg.indirect = ralloc(state->shader, nir_src); 17827c6e3e4ca665c528b94937b1b0a7c10de515d89Jason Ekstrand nir_src_copy(src.reg.indirect, &deref_array->indirect, 1794e9b3765947f2c98d618082cd449db2e319bd887Jason Ekstrand state->shader); 180aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand } else { 1814e9b3765947f2c98d618082cd449db2e319bd887Jason Ekstrand nir_alu_instr *add = nir_alu_instr_create(state->shader, 182aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand nir_op_iadd); 183aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand add->src[0].src = *src.reg.indirect; 1848c8fc5f8336c8c79e5890265ae6c03271aa94075Jason Ekstrand nir_src_copy(&add->src[1].src, &deref_array->indirect, add); 185aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand add->dest.write_mask = 1; 1863124ce699bb3844e793f00e00bfbea5c91744f90Connor Abbott nir_ssa_dest_init(&add->instr, &add->dest.dest, 1, 32, NULL); 187aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand nir_instr_insert_before(instr, &add->instr); 188aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 189aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand src.reg.indirect->is_ssa = true; 190aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand src.reg.indirect->ssa = &add->dest.dest.ssa; 191aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand } 192aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand } 193aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand } 194aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 195aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand return src; 196aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand} 197aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 198aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrandstatic bool 19915264133d7f3b2c63839ae20f45243b62d9f8eddConnor Abbottlower_locals_to_regs_block(nir_block *block, 20015264133d7f3b2c63839ae20f45243b62d9f8eddConnor Abbott struct locals_to_regs_state *state) 201aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand{ 202707e72f13bb78869ee95d3286980bf1709cba6cfJason Ekstrand nir_foreach_instr_safe(instr, block) { 203aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand if (instr->type != nir_instr_type_intrinsic) 204aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand continue; 205aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 206aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand nir_intrinsic_instr *intrin = nir_instr_as_intrinsic(instr); 207aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 208aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand switch (intrin->intrinsic) { 20927663dbe8edfb7583d9d8fc3704a04a5c837fe05Jason Ekstrand case nir_intrinsic_load_var: { 210aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand if (intrin->variables[0]->var->data.mode != nir_var_local) 211aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand continue; 212aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 2134e9b3765947f2c98d618082cd449db2e319bd887Jason Ekstrand nir_alu_instr *mov = nir_alu_instr_create(state->shader, nir_op_imov); 214aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand mov->src[0].src = get_deref_reg_src(intrin->variables[0], 215aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand &intrin->instr, state); 21627663dbe8edfb7583d9d8fc3704a04a5c837fe05Jason Ekstrand mov->dest.write_mask = (1 << intrin->num_components) - 1; 217aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand if (intrin->dest.is_ssa) { 218534a4ec82f09fa37e2134c994082e71c79b7d5ecEric Anholt nir_ssa_dest_init(&mov->instr, &mov->dest.dest, 2193124ce699bb3844e793f00e00bfbea5c91744f90Connor Abbott intrin->num_components, 2203124ce699bb3844e793f00e00bfbea5c91744f90Connor Abbott intrin->dest.ssa.bit_size, NULL); 221f88c6a49970ee35d29840e03fd6512a7acd815fbJason Ekstrand nir_ssa_def_rewrite_uses(&intrin->dest.ssa, 222a4aa25be1e0a27b1a6a6b0bcf576beb9dfe1ea7aJason Ekstrand nir_src_for_ssa(&mov->dest.dest.ssa)); 223aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand } else { 2248c8fc5f8336c8c79e5890265ae6c03271aa94075Jason Ekstrand nir_dest_copy(&mov->dest.dest, &intrin->dest, &mov->instr); 225aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand } 226aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand nir_instr_insert_before(&intrin->instr, &mov->instr); 227aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 228aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand nir_instr_remove(&intrin->instr); 229cfae0f8a3ae98ae10bcf9e4f1429f66165a17d95Kenneth Graunke state->progress = true; 230aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand break; 231aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand } 232aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 23327663dbe8edfb7583d9d8fc3704a04a5c837fe05Jason Ekstrand case nir_intrinsic_store_var: { 234aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand if (intrin->variables[0]->var->data.mode != nir_var_local) 235aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand continue; 236aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 237aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand nir_src reg_src = get_deref_reg_src(intrin->variables[0], 238aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand &intrin->instr, state); 239aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 2404e9b3765947f2c98d618082cd449db2e319bd887Jason Ekstrand nir_alu_instr *mov = nir_alu_instr_create(state->shader, nir_op_imov); 2418c8fc5f8336c8c79e5890265ae6c03271aa94075Jason Ekstrand nir_src_copy(&mov->src[0].src, &intrin->src[0], mov); 242ced8d3e7730777ea8a264d22f83b43f7b3a5c433Rob Clark mov->dest.write_mask = nir_intrinsic_write_mask(intrin); 243aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand mov->dest.dest.is_ssa = false; 244aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand mov->dest.dest.reg.reg = reg_src.reg.reg; 245aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand mov->dest.dest.reg.base_offset = reg_src.reg.base_offset; 246aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand mov->dest.dest.reg.indirect = reg_src.reg.indirect; 247aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 248aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand nir_instr_insert_before(&intrin->instr, &mov->instr); 249aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 250aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand nir_instr_remove(&intrin->instr); 251cfae0f8a3ae98ae10bcf9e4f1429f66165a17d95Kenneth Graunke state->progress = true; 252aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand break; 253aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand } 254aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 255aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand case nir_intrinsic_copy_var: 256aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand unreachable("There should be no copies whatsoever at this point"); 257aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand break; 258aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 259aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand default: 260aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand continue; 261aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand } 262aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand } 263aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 264aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand return true; 265aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand} 266aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 267cfae0f8a3ae98ae10bcf9e4f1429f66165a17d95Kenneth Graunkestatic bool 268aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrandnir_lower_locals_to_regs_impl(nir_function_impl *impl) 269aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand{ 270aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand struct locals_to_regs_state state; 271aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 272237f2f2d8b45d9d956102eec6f9be63193e5269bJason Ekstrand state.shader = impl->function->shader; 273aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand state.impl = impl; 274cfae0f8a3ae98ae10bcf9e4f1429f66165a17d95Kenneth Graunke state.progress = false; 275aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand state.regs_table = _mesa_hash_table_create(NULL, hash_deref, derefs_equal); 27606f3c98b9da35b5f5c02bd30599fbde57a19520eJason Ekstrand 27706f3c98b9da35b5f5c02bd30599fbde57a19520eJason Ekstrand nir_metadata_require(impl, nir_metadata_dominance); 278aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 27915264133d7f3b2c63839ae20f45243b62d9f8eddConnor Abbott nir_foreach_block(block, impl) { 28015264133d7f3b2c63839ae20f45243b62d9f8eddConnor Abbott lower_locals_to_regs_block(block, &state); 28115264133d7f3b2c63839ae20f45243b62d9f8eddConnor Abbott } 282aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 283295faf9462cba88250d8581f65611996eba5e389Jason Ekstrand nir_metadata_preserve(impl, nir_metadata_block_index | 284295faf9462cba88250d8581f65611996eba5e389Jason Ekstrand nir_metadata_dominance); 285295faf9462cba88250d8581f65611996eba5e389Jason Ekstrand 286aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand _mesa_hash_table_destroy(state.regs_table, NULL); 287cfae0f8a3ae98ae10bcf9e4f1429f66165a17d95Kenneth Graunke 288cfae0f8a3ae98ae10bcf9e4f1429f66165a17d95Kenneth Graunke return state.progress; 289aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand} 290aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand 291cfae0f8a3ae98ae10bcf9e4f1429f66165a17d95Kenneth Graunkebool 292aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrandnir_lower_locals_to_regs(nir_shader *shader) 293aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand{ 294cfae0f8a3ae98ae10bcf9e4f1429f66165a17d95Kenneth Graunke bool progress = false; 295cfae0f8a3ae98ae10bcf9e4f1429f66165a17d95Kenneth Graunke 2969464d8c49813aba77285e7465b96e92a91ed327cJason Ekstrand nir_foreach_function(function, shader) { 297237f2f2d8b45d9d956102eec6f9be63193e5269bJason Ekstrand if (function->impl) 298237f2f2d8b45d9d956102eec6f9be63193e5269bJason Ekstrand progress = nir_lower_locals_to_regs_impl(function->impl) || progress; 299aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand } 300cfae0f8a3ae98ae10bcf9e4f1429f66165a17d95Kenneth Graunke 301cfae0f8a3ae98ae10bcf9e4f1429f66165a17d95Kenneth Graunke return progress; 302aff431293b441368acd907104c8acd1b117bfaf8Jason Ekstrand} 303