AlexIndustrial/mesa
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

nir: allow cloning indirect array derefs in nir_clone_deref_instr

Browse Source 
but only if cloning within the same shader. This will be used to fix
nir_lower_io_to_temporaries.

Reviewed-by: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/32344>
This commit is contained in:
Marek Olšák
2024-11-25 20:40:22 -05:00
committed by Marge Bot
parent a3516dafc9
commit c23abb12e8
1 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/compiler/nir/nir_linking_helpers.c
+12 -3
View File
@@ -1202,9 +1202,18 @@ nir_clone_deref_instr(nir_builder *b, nir_variable *var,
*/
switch (deref->deref_type) {
case nir_deref_type_array: {
nir_load_const_instr *index =
nir_instr_as_load_const(deref->arr.index.ssa->parent_instr);
return nir_build_deref_array_imm(b, parent, index->value->i64);
if (b->shader ==
nir_cf_node_get_function(&deref->instr.block->cf_node)->function->shader) {
/* Cloning within the same shader. */
return nir_build_deref_array(b, parent, deref->arr.index.ssa);
} else {
/* Cloning to a different shader. The index must be constant because
* we don't implement cloning the index SSA here.
*/
nir_load_const_instr *index =
nir_instr_as_load_const(deref->arr.index.ssa->parent_instr);
return nir_build_deref_array_imm(b, parent, index->value->i64);
}
}
case nir_deref_type_ptr_as_array: {
nir_load_const_instr *index =