*/
static unsigned
insert_mov(nir_alu_instr *vec, unsigned start_channel,
- unsigned start_src_idx, void *mem_ctx)
+ unsigned start_src_idx, nir_shader *shader)
{
unsigned src_idx = start_src_idx;
assert(src_idx < nir_op_infos[vec->op].num_inputs);
- nir_alu_instr *mov = nir_alu_instr_create(mem_ctx, nir_op_imov);
+ nir_alu_instr *mov = nir_alu_instr_create(shader, nir_op_imov);
nir_alu_src_copy(&mov->src[0], &vec->src[src_idx], mov);
nir_alu_dest_copy(&mov->dest, &vec->dest, mov);
}
static bool
-lower_vec_to_movs_block(nir_block *block, void *mem_ctx)
+lower_vec_to_movs_block(nir_block *block, void *shader)
{
nir_foreach_instr_safe(block, instr) {
if (instr->type != nir_instr_type_alu)
continue;
if (src_matches_dest_reg(&vec->dest.dest, &vec->src[src_idx].src)) {
- finished_write_mask |= insert_mov(vec, i, src_idx, mem_ctx);
+ finished_write_mask |= insert_mov(vec, i, src_idx, shader);
break;
}
src_idx++;
continue;
if (!(finished_write_mask & (1 << i)))
- finished_write_mask |= insert_mov(vec, i, src_idx, mem_ctx);
+ finished_write_mask |= insert_mov(vec, i, src_idx, shader);
src_idx++;
}
static void
nir_lower_vec_to_movs_impl(nir_function_impl *impl)
{
- nir_foreach_block(impl, lower_vec_to_movs_block, ralloc_parent(impl));
+ nir_shader *shader = impl->overload->function->shader;
+
+ nir_foreach_block(impl, lower_vec_to_movs_block, shader);
}
void