etnaviv: move liveness related stuff into own file

Signed-off-by: Christian Gmeiner <christian.gmeiner@gmail.com>
Acked-by: Jonathan Marek <jonathan@marek.ca>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/5690>
This commit is contained in:
Christian Gmeiner
2020-06-29 16:00:29 +02:00
committed by Marge Bot
parent 9ae96d32dd
commit 0f025e8b81
6 changed files with 424 additions and 391 deletions

View File

@@ -19,6 +19,7 @@ C_SOURCES := \
etnaviv_compiler.h \
etnaviv_compiler_nir.c \
etnaviv_compiler_nir_emit.h \
etnaviv_compiler_nir_liveness.c \
etnaviv_compiler_tgsi.c \
etnaviv_context.c \
etnaviv_context.h \

View File

@@ -40,7 +40,6 @@
#include "util/u_memory.h"
#include "util/register_allocate.h"
#include "compiler/nir/nir_builder.h"
#include "compiler/nir/nir_worklist.h"
#include "tgsi/tgsi_strings.h"
#include "util/u_half.h"

View File

@@ -27,10 +27,181 @@
#ifndef H_ETNAVIV_COMPILER_NIR
#define H_ETNAVIV_COMPILER_NIR
#include "compiler/nir/nir.h"
#define compile_error(ctx, args...) ({ \
printf(args); \
ctx->error = true; \
assert(0); \
})
enum {
BYPASS_DST = 1,
BYPASS_SRC = 2,
};
static inline bool is_sysval(nir_instr *instr)
{
if (instr->type != nir_instr_type_intrinsic)
return false;
nir_intrinsic_instr *intr = nir_instr_as_intrinsic(instr);
return intr->intrinsic == nir_intrinsic_load_front_face ||
intr->intrinsic == nir_intrinsic_load_frag_coord;
}
/* get unique ssa/reg index for nir_src */
static inline unsigned
src_index(nir_function_impl *impl, nir_src *src)
{
return src->is_ssa ? src->ssa->index : (src->reg.reg->index + impl->ssa_alloc);
}
/* get unique ssa/reg index for nir_dest */
static inline unsigned
dest_index(nir_function_impl *impl, nir_dest *dest)
{
return dest->is_ssa ? dest->ssa.index : (dest->reg.reg->index + impl->ssa_alloc);
}
static inline void
update_swiz_mask(nir_alu_instr *alu, nir_dest *dest, unsigned *swiz, unsigned *mask)
{
if (!swiz)
return;
bool is_vec = dest != NULL;
unsigned swizzle = 0, write_mask = 0;
for (unsigned i = 0; i < 4; i++) {
/* channel not written */
if (!(alu->dest.write_mask & (1 << i)))
continue;
/* src is different (only check for vecN) */
if (is_vec && alu->src[i].src.ssa != &dest->ssa)
continue;
unsigned src_swiz = is_vec ? alu->src[i].swizzle[0] : alu->src[0].swizzle[i];
swizzle |= (*swiz >> src_swiz * 2 & 3) << i * 2;
/* this channel isn't written through this chain */
if (*mask & (1 << src_swiz))
write_mask |= 1 << i;
}
*swiz = swizzle;
*mask = write_mask;
}
static nir_dest *
real_dest(nir_dest *dest, unsigned *swiz, unsigned *mask)
{
if (!dest || !dest->is_ssa)
return dest;
bool can_bypass_src = !list_length(&dest->ssa.if_uses);
nir_instr *p_instr = dest->ssa.parent_instr;
/* if used by a vecN, the "real" destination becomes the vecN destination
* lower_alu guarantees that values used by a vecN are only used by that vecN
* we can apply the same logic to movs in a some cases too
*/
nir_foreach_use(use_src, &dest->ssa) {
nir_instr *instr = use_src->parent_instr;
/* src bypass check: for now only deal with tex src mov case
* note: for alu don't bypass mov for multiple uniform sources
*/
switch (instr->type) {
case nir_instr_type_tex:
if (p_instr->type == nir_instr_type_alu &&
nir_instr_as_alu(p_instr)->op == nir_op_mov) {
break;
}
default:
can_bypass_src = false;
break;
}
if (instr->type != nir_instr_type_alu)
continue;
nir_alu_instr *alu = nir_instr_as_alu(instr);
switch (alu->op) {
case nir_op_vec2:
case nir_op_vec3:
case nir_op_vec4:
assert(list_length(&dest->ssa.if_uses) == 0);
nir_foreach_use(use_src, &dest->ssa)
assert(use_src->parent_instr == instr);
update_swiz_mask(alu, dest, swiz, mask);
break;
case nir_op_mov: {
switch (dest->ssa.parent_instr->type) {
case nir_instr_type_alu:
case nir_instr_type_tex:
break;
default:
continue;
}
if (list_length(&dest->ssa.if_uses) || list_length(&dest->ssa.uses) > 1)
continue;
update_swiz_mask(alu, NULL, swiz, mask);
break;
};
default:
continue;
}
assert(!(instr->pass_flags & BYPASS_SRC));
instr->pass_flags |= BYPASS_DST;
return real_dest(&alu->dest.dest, swiz, mask);
}
if (can_bypass_src && !(p_instr->pass_flags & BYPASS_DST)) {
p_instr->pass_flags |= BYPASS_SRC;
return NULL;
}
return dest;
}
/* if instruction dest needs a register, return nir_dest for it */
static inline nir_dest *
dest_for_instr(nir_instr *instr)
{
nir_dest *dest = NULL;
switch (instr->type) {
case nir_instr_type_alu:
dest = &nir_instr_as_alu(instr)->dest.dest;
break;
case nir_instr_type_tex:
dest = &nir_instr_as_tex(instr)->dest;
break;
case nir_instr_type_intrinsic: {
nir_intrinsic_instr *intr = nir_instr_as_intrinsic(instr);
if (intr->intrinsic == nir_intrinsic_load_uniform ||
intr->intrinsic == nir_intrinsic_load_ubo ||
intr->intrinsic == nir_intrinsic_load_input ||
intr->intrinsic == nir_intrinsic_load_instance_id)
dest = &intr->dest;
} break;
case nir_instr_type_deref:
return NULL;
default:
break;
}
return real_dest(dest, NULL, NULL);
}
struct live_def {
nir_instr *instr;
nir_dest *dest; /* cached dest_for_instr */
unsigned live_start, live_end; /* live range */
};
unsigned
etna_live_defs(nir_function_impl *impl, struct live_def *defs, unsigned *live_map);
#endif

View File

@@ -30,7 +30,6 @@
#include "compiler/nir/nir.h"
#include "compiler/nir/nir_builder.h"
#include "compiler/nir/nir_worklist.h"
#include "util/register_allocate.h"
#define ALU_SWIZ(s) INST_SWIZ((s)->swizzle[0], (s)->swizzle[1], (s)->swizzle[2], (s)->swizzle[3])
@@ -43,11 +42,6 @@
typedef struct etna_inst_dst hw_dst;
typedef struct etna_inst_src hw_src;
enum {
BYPASS_DST = 1,
BYPASS_SRC = 2,
};
struct state {
struct etna_compile *c;
@@ -72,16 +66,6 @@ src_swizzle(hw_src src, unsigned swizzle)
return src;
}
static inline bool is_sysval(nir_instr *instr)
{
if (instr->type != nir_instr_type_intrinsic)
return false;
nir_intrinsic_instr *intr = nir_instr_as_intrinsic(instr);
return intr->intrinsic == nir_intrinsic_load_front_face ||
intr->intrinsic == nir_intrinsic_load_frag_coord;
}
/* constants are represented as 64-bit ints
* 32-bit for the value and 32-bit for the type (imm, uniform, etc)
*/
@@ -329,20 +313,6 @@ static inline int reg_get_class(int virt_reg)
return 0;
}
/* get unique ssa/reg index for nir_src */
static unsigned
src_index(nir_function_impl *impl, nir_src *src)
{
return src->is_ssa ? src->ssa->index : (src->reg.reg->index + impl->ssa_alloc);
}
/* get unique ssa/reg index for nir_dest */
static unsigned
dest_index(nir_function_impl *impl, nir_dest *dest)
{
return dest->is_ssa ? dest->ssa.index : (dest->reg.reg->index + impl->ssa_alloc);
}
/* nir_src to allocated register */
static hw_src
ra_src(struct state *state, nir_src *src)
@@ -403,32 +373,6 @@ get_src(struct state *state, nir_src *src)
return SRC_DISABLE;
}
static void
update_swiz_mask(nir_alu_instr *alu, nir_dest *dest, unsigned *swiz, unsigned *mask)
{
if (!swiz)
return;
bool is_vec = dest != NULL;
unsigned swizzle = 0, write_mask = 0;
for (unsigned i = 0; i < 4; i++) {
/* channel not written */
if (!(alu->dest.write_mask & (1 << i)))
continue;
/* src is different (only check for vecN) */
if (is_vec && alu->src[i].src.ssa != &dest->ssa)
continue;
unsigned src_swiz = is_vec ? alu->src[i].swizzle[0] : alu->src[0].swizzle[i];
swizzle |= (*swiz >> src_swiz * 2 & 3) << i * 2;
/* this channel isn't written through this chain */
if (*mask & (1 << src_swiz))
write_mask |= 1 << i;
}
*swiz = swizzle;
*mask = write_mask;
}
static bool
vec_dest_has_swizzle(nir_alu_instr *vec, nir_ssa_def *ssa)
{
@@ -461,82 +405,6 @@ vec_dest_has_swizzle(nir_alu_instr *vec, nir_ssa_def *ssa)
return false;
}
static nir_dest *
real_dest(nir_dest *dest, unsigned *swiz, unsigned *mask)
{
if (!dest || !dest->is_ssa)
return dest;
bool can_bypass_src = !list_length(&dest->ssa.if_uses);
nir_instr *p_instr = dest->ssa.parent_instr;
/* if used by a vecN, the "real" destination becomes the vecN destination
* lower_alu guarantees that values used by a vecN are only used by that vecN
* we can apply the same logic to movs in a some cases too
*/
nir_foreach_use(use_src, &dest->ssa) {
nir_instr *instr = use_src->parent_instr;
/* src bypass check: for now only deal with tex src mov case
* note: for alu don't bypass mov for multiple uniform sources
*/
switch (instr->type) {
case nir_instr_type_tex:
if (p_instr->type == nir_instr_type_alu &&
nir_instr_as_alu(p_instr)->op == nir_op_mov) {
break;
}
default:
can_bypass_src = false;
break;
}
if (instr->type != nir_instr_type_alu)
continue;
nir_alu_instr *alu = nir_instr_as_alu(instr);
switch (alu->op) {
case nir_op_vec2:
case nir_op_vec3:
case nir_op_vec4:
assert(list_length(&dest->ssa.if_uses) == 0);
nir_foreach_use(use_src, &dest->ssa)
assert(use_src->parent_instr == instr);
update_swiz_mask(alu, dest, swiz, mask);
break;
case nir_op_mov: {
switch (dest->ssa.parent_instr->type) {
case nir_instr_type_alu:
case nir_instr_type_tex:
break;
default:
continue;
}
if (list_length(&dest->ssa.if_uses) || list_length(&dest->ssa.uses) > 1)
continue;
update_swiz_mask(alu, NULL, swiz, mask);
break;
};
default:
continue;
}
assert(!(instr->pass_flags & BYPASS_SRC));
instr->pass_flags |= BYPASS_DST;
return real_dest(&alu->dest.dest, swiz, mask);
}
if (can_bypass_src && !(p_instr->pass_flags & BYPASS_DST)) {
p_instr->pass_flags |= BYPASS_SRC;
return NULL;
}
return dest;
}
/* get allocated dest register for nir_dest
* *p_swiz tells how the components need to be placed into register
*/
@@ -558,263 +426,6 @@ ra_dest(struct state *state, nir_dest *dest, unsigned *p_swiz)
};
}
/* if instruction dest needs a register, return nir_dest for it */
static nir_dest *
dest_for_instr(nir_instr *instr)
{
nir_dest *dest = NULL;
switch (instr->type) {
case nir_instr_type_alu:
dest = &nir_instr_as_alu(instr)->dest.dest;
break;
case nir_instr_type_tex:
dest = &nir_instr_as_tex(instr)->dest;
break;
case nir_instr_type_intrinsic: {
nir_intrinsic_instr *intr = nir_instr_as_intrinsic(instr);
if (intr->intrinsic == nir_intrinsic_load_uniform ||
intr->intrinsic == nir_intrinsic_load_ubo ||
intr->intrinsic == nir_intrinsic_load_input ||
intr->intrinsic == nir_intrinsic_load_instance_id)
dest = &intr->dest;
} break;
case nir_instr_type_deref:
return NULL;
default:
break;
}
return real_dest(dest, NULL, NULL);
}
struct live_def {
nir_instr *instr;
nir_dest *dest; /* cached dest_for_instr */
unsigned live_start, live_end; /* live range */
};
static void
range_include(struct live_def *def, unsigned index)
{
if (def->live_start > index)
def->live_start = index;
if (def->live_end < index)
def->live_end = index;
}
struct live_defs_state {
unsigned num_defs;
unsigned bitset_words;
nir_function_impl *impl;
nir_block *block; /* current block pointer */
unsigned index; /* current live index */
struct live_def *defs;
unsigned *live_map; /* to map ssa/reg index into defs array */
nir_block_worklist worklist;
};
static bool
init_liveness_block(nir_block *block,
struct live_defs_state *state)
{
block->live_in = reralloc(block, block->live_in, BITSET_WORD,
state->bitset_words);
memset(block->live_in, 0, state->bitset_words * sizeof(BITSET_WORD));
block->live_out = reralloc(block, block->live_out, BITSET_WORD,
state->bitset_words);
memset(block->live_out, 0, state->bitset_words * sizeof(BITSET_WORD));
nir_block_worklist_push_head(&state->worklist, block);
return true;
}
static bool
set_src_live(nir_src *src, void *void_state)
{
struct live_defs_state *state = void_state;
if (src->is_ssa) {
nir_instr *instr = src->ssa->parent_instr;
if (is_sysval(instr) || instr->type == nir_instr_type_deref)
return true;
switch (instr->type) {
case nir_instr_type_load_const:
case nir_instr_type_ssa_undef:
return true;
case nir_instr_type_alu: {
/* alu op bypass */
nir_alu_instr *alu = nir_instr_as_alu(instr);
if (instr->pass_flags & BYPASS_SRC) {
for (unsigned i = 0; i < nir_op_infos[alu->op].num_inputs; i++)
set_src_live(&alu->src[i].src, state);
return true;
}
} break;
default:
break;
}
}
unsigned i = state->live_map[src_index(state->impl, src)];
assert(i != ~0u);
BITSET_SET(state->block->live_in, i);
range_include(&state->defs[i], state->index);
return true;
}
static bool
propagate_across_edge(nir_block *pred, nir_block *succ,
struct live_defs_state *state)
{
BITSET_WORD progress = 0;
for (unsigned i = 0; i < state->bitset_words; ++i) {
progress |= succ->live_in[i] & ~pred->live_out[i];
pred->live_out[i] |= succ->live_in[i];
}
return progress != 0;
}
static unsigned
live_defs(nir_function_impl *impl, struct live_def *defs, unsigned *live_map)
{
struct live_defs_state state;
unsigned block_live_index[impl->num_blocks + 1];
state.impl = impl;
state.defs = defs;
state.live_map = live_map;
state.num_defs = 0;
nir_foreach_block(block, impl) {
block_live_index[block->index] = state.num_defs;
nir_foreach_instr(instr, block) {
nir_dest *dest = dest_for_instr(instr);
if (!dest)
continue;
unsigned idx = dest_index(impl, dest);
/* register is already in defs */
if (live_map[idx] != ~0u)
continue;
defs[state.num_defs] = (struct live_def) {instr, dest, state.num_defs, 0};
/* input live from the start */
if (instr->type == nir_instr_type_intrinsic) {
nir_intrinsic_instr *intr = nir_instr_as_intrinsic(instr);
if (intr->intrinsic == nir_intrinsic_load_input ||
intr->intrinsic == nir_intrinsic_load_instance_id)
defs[state.num_defs].live_start = 0;
}
live_map[idx] = state.num_defs;
state.num_defs++;
}
}
block_live_index[impl->num_blocks] = state.num_defs;
nir_block_worklist_init(&state.worklist, impl->num_blocks, NULL);
/* We now know how many unique ssa definitions we have and we can go
* ahead and allocate live_in and live_out sets and add all of the
* blocks to the worklist.
*/
state.bitset_words = BITSET_WORDS(state.num_defs);
nir_foreach_block(block, impl) {
init_liveness_block(block, &state);
}
/* We're now ready to work through the worklist and update the liveness
* sets of each of the blocks. By the time we get to this point, every
* block in the function implementation has been pushed onto the
* worklist in reverse order. As long as we keep the worklist
* up-to-date as we go, everything will get covered.
*/
while (!nir_block_worklist_is_empty(&state.worklist)) {
/* We pop them off in the reverse order we pushed them on. This way
* the first walk of the instructions is backwards so we only walk
* once in the case of no control flow.
*/
nir_block *block = nir_block_worklist_pop_head(&state.worklist);
state.block = block;
memcpy(block->live_in, block->live_out,
state.bitset_words * sizeof(BITSET_WORD));
state.index = block_live_index[block->index + 1];
nir_if *following_if = nir_block_get_following_if(block);
if (following_if)
set_src_live(&following_if->condition, &state);
nir_foreach_instr_reverse(instr, block) {
/* when we come across the next "live" instruction, decrement index */
if (state.index && instr == defs[state.index - 1].instr) {
state.index--;
/* the only source of writes to registers is phis:
* we don't expect any partial write_mask alus
* so clearing live_in here is OK
*/
BITSET_CLEAR(block->live_in, state.index);
}
/* don't set_src_live for not-emitted instructions */
if (instr->pass_flags)
continue;
unsigned index = state.index;
/* output live till the end */
if (instr->type == nir_instr_type_intrinsic) {
nir_intrinsic_instr *intr = nir_instr_as_intrinsic(instr);
if (intr->intrinsic == nir_intrinsic_store_deref)
state.index = ~0u;
}
nir_foreach_src(instr, set_src_live, &state);
state.index = index;
}
assert(state.index == block_live_index[block->index]);
/* Walk over all of the predecessors of the current block updating
* their live in with the live out of this one. If anything has
* changed, add the predecessor to the work list so that we ensure
* that the new information is used.
*/
set_foreach(block->predecessors, entry) {
nir_block *pred = (nir_block *)entry->key;
if (propagate_across_edge(pred, block, &state))
nir_block_worklist_push_tail(&state.worklist, pred);
}
}
nir_block_worklist_fini(&state.worklist);
/* apply live_in/live_out to ranges */
nir_foreach_block(block, impl) {
int i;
BITSET_FOREACH_SET(i, block->live_in, state.num_defs)
range_include(&state.defs[i], block_live_index[block->index]);
BITSET_FOREACH_SET(i, block->live_out, state.num_defs)
range_include(&state.defs[i], block_live_index[block->index + 1]);
}
return state.num_defs;
}
/* precomputed by register_allocate */
static unsigned int *q_values[] = {
(unsigned int[]) {1, 2, 3, 4, 2, 2, 3, },
@@ -872,7 +483,7 @@ ra_assign(struct state *state, nir_shader *shader)
memset(live_map, 0xff, sizeof(unsigned) * max_nodes);
struct live_def *defs = rzalloc_array(NULL, struct live_def, max_nodes);
unsigned num_nodes = live_defs(impl, defs, live_map);
unsigned num_nodes = etna_live_defs(impl, defs, live_map);
struct ra_graph *g = ra_alloc_interference_graph(regs, num_nodes);
/* set classes from num_components */

View File

@@ -0,0 +1,250 @@
/*
* Copyright (c) 2019 Zodiac Inflight Innovations
*
* Permission is hereby granted, free of charge, to any person obtaining a
* copy of this software and associated documentation files (the "Software"),
* to deal in the Software without restriction, including without limitation
* the rights to use, copy, modify, merge, publish, distribute, sub license,
* and/or sell copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice (including the
* next paragraph) shall be included in all copies or substantial portions
* of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
* DEALINGS IN THE SOFTWARE.
*
* Authors:
* Jonathan Marek <jonathan@marek.ca>
*/
#include "etnaviv_compiler_nir.h"
#include "compiler/nir/nir_worklist.h"
static void
range_include(struct live_def *def, unsigned index)
{
if (def->live_start > index)
def->live_start = index;
if (def->live_end < index)
def->live_end = index;
}
struct live_defs_state {
unsigned num_defs;
unsigned bitset_words;
nir_function_impl *impl;
nir_block *block; /* current block pointer */
unsigned index; /* current live index */
struct live_def *defs;
unsigned *live_map; /* to map ssa/reg index into defs array */
nir_block_worklist worklist;
};
static bool
init_liveness_block(nir_block *block,
struct live_defs_state *state)
{
block->live_in = reralloc(block, block->live_in, BITSET_WORD,
state->bitset_words);
memset(block->live_in, 0, state->bitset_words * sizeof(BITSET_WORD));
block->live_out = reralloc(block, block->live_out, BITSET_WORD,
state->bitset_words);
memset(block->live_out, 0, state->bitset_words * sizeof(BITSET_WORD));
nir_block_worklist_push_head(&state->worklist, block);
return true;
}
static bool
set_src_live(nir_src *src, void *void_state)
{
struct live_defs_state *state = void_state;
if (src->is_ssa) {
nir_instr *instr = src->ssa->parent_instr;
if (is_sysval(instr) || instr->type == nir_instr_type_deref)
return true;
switch (instr->type) {
case nir_instr_type_load_const:
case nir_instr_type_ssa_undef:
return true;
case nir_instr_type_alu: {
/* alu op bypass */
nir_alu_instr *alu = nir_instr_as_alu(instr);
if (instr->pass_flags & BYPASS_SRC) {
for (unsigned i = 0; i < nir_op_infos[alu->op].num_inputs; i++)
set_src_live(&alu->src[i].src, state);
return true;
}
} break;
default:
break;
}
}
unsigned i = state->live_map[src_index(state->impl, src)];
assert(i != ~0u);
BITSET_SET(state->block->live_in, i);
range_include(&state->defs[i], state->index);
return true;
}
static bool
propagate_across_edge(nir_block *pred, nir_block *succ,
struct live_defs_state *state)
{
BITSET_WORD progress = 0;
for (unsigned i = 0; i < state->bitset_words; ++i) {
progress |= succ->live_in[i] & ~pred->live_out[i];
pred->live_out[i] |= succ->live_in[i];
}
return progress != 0;
}
unsigned
etna_live_defs(nir_function_impl *impl, struct live_def *defs, unsigned *live_map)
{
struct live_defs_state state;
unsigned block_live_index[impl->num_blocks + 1];
state.impl = impl;
state.defs = defs;
state.live_map = live_map;
state.num_defs = 0;
nir_foreach_block(block, impl) {
block_live_index[block->index] = state.num_defs;
nir_foreach_instr(instr, block) {
nir_dest *dest = dest_for_instr(instr);
if (!dest)
continue;
unsigned idx = dest_index(impl, dest);
/* register is already in defs */
if (live_map[idx] != ~0u)
continue;
defs[state.num_defs] = (struct live_def) {instr, dest, state.num_defs, 0};
/* input live from the start */
if (instr->type == nir_instr_type_intrinsic) {
nir_intrinsic_instr *intr = nir_instr_as_intrinsic(instr);
if (intr->intrinsic == nir_intrinsic_load_input ||
intr->intrinsic == nir_intrinsic_load_instance_id)
defs[state.num_defs].live_start = 0;
}
live_map[idx] = state.num_defs;
state.num_defs++;
}
}
block_live_index[impl->num_blocks] = state.num_defs;
nir_block_worklist_init(&state.worklist, impl->num_blocks, NULL);
/* We now know how many unique ssa definitions we have and we can go
* ahead and allocate live_in and live_out sets and add all of the
* blocks to the worklist.
*/
state.bitset_words = BITSET_WORDS(state.num_defs);
nir_foreach_block(block, impl) {
init_liveness_block(block, &state);
}
/* We're now ready to work through the worklist and update the liveness
* sets of each of the blocks. By the time we get to this point, every
* block in the function implementation has been pushed onto the
* worklist in reverse order. As long as we keep the worklist
* up-to-date as we go, everything will get covered.
*/
while (!nir_block_worklist_is_empty(&state.worklist)) {
/* We pop them off in the reverse order we pushed them on. This way
* the first walk of the instructions is backwards so we only walk
* once in the case of no control flow.
*/
nir_block *block = nir_block_worklist_pop_head(&state.worklist);
state.block = block;
memcpy(block->live_in, block->live_out,
state.bitset_words * sizeof(BITSET_WORD));
state.index = block_live_index[block->index + 1];
nir_if *following_if = nir_block_get_following_if(block);
if (following_if)
set_src_live(&following_if->condition, &state);
nir_foreach_instr_reverse(instr, block) {
/* when we come across the next "live" instruction, decrement index */
if (state.index && instr == defs[state.index - 1].instr) {
state.index--;
/* the only source of writes to registers is phis:
* we don't expect any partial write_mask alus
* so clearing live_in here is OK
*/
BITSET_CLEAR(block->live_in, state.index);
}
/* don't set_src_live for not-emitted instructions */
if (instr->pass_flags)
continue;
unsigned index = state.index;
/* output live till the end */
if (instr->type == nir_instr_type_intrinsic) {
nir_intrinsic_instr *intr = nir_instr_as_intrinsic(instr);
if (intr->intrinsic == nir_intrinsic_store_deref)
state.index = ~0u;
}
nir_foreach_src(instr, set_src_live, &state);
state.index = index;
}
assert(state.index == block_live_index[block->index]);
/* Walk over all of the predecessors of the current block updating
* their live in with the live out of this one. If anything has
* changed, add the predecessor to the work list so that we ensure
* that the new information is used.
*/
set_foreach(block->predecessors, entry) {
nir_block *pred = (nir_block *)entry->key;
if (propagate_across_edge(pred, block, &state))
nir_block_worklist_push_tail(&state.worklist, pred);
}
}
nir_block_worklist_fini(&state.worklist);
/* apply live_in/live_out to ranges */
nir_foreach_block(block, impl) {
int i;
BITSET_FOREACH_SET(i, block->live_in, state.num_defs)
range_include(&state.defs[i], block_live_index[block->index]);
BITSET_FOREACH_SET(i, block->live_out, state.num_defs)
range_include(&state.defs[i], block_live_index[block->index + 1]);
}
return state.num_defs;
}

View File

@@ -38,6 +38,7 @@ files_etnaviv = files(
'etnaviv_compiler.h',
'etnaviv_compiler_nir.c',
'etnaviv_compiler_nir_emit.h',
'etnaviv_compiler_nir_liveness.c',
'etnaviv_compiler_tgsi.c',
'etnaviv_context.c',
'etnaviv_context.h',