- nir_ssa_def *new_offset = nir_iadd(b, offset_src->ssa,
- &offset_load->dest.ssa);
+ nir_ssa_def *old_offset = nir_ssa_for_src(b, *offset_src, 1);
+ nir_ssa_def *new_offset = nir_iadd(b, old_offset, &offset_load->dest.ssa);
+ nir_instr_rewrite_src(&intrin->instr, offset_src,
+ nir_src_for_ssa(new_offset));