nir/lower_shader_calls: fix store_scratch write_mask

Signed-off-by: Rhys Perry <pendingchaos02@gmail.com>
Reviewed-by: Timur Kristóf <timur.kristof@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/14447>
This commit is contained in:
Rhys Perry 2022-01-07 10:58:19 +00:00 committed by Marge Bot
parent d799a4be27
commit b00138090e

View file

@ -273,7 +273,8 @@ spill_fill(nir_builder *before, nir_builder *after, nir_ssa_def *def, unsigned o
switch(address_format) {
case nir_address_format_32bit_offset:
nir_store_scratch(before, def, nir_imm_int(before, offset),
.align_mul = MIN2(comp_size, stack_alignment), .write_mask = ~0);
.align_mul = MIN2(comp_size, stack_alignment),
.write_mask = BITFIELD_MASK(def->num_components));
def = nir_load_scratch(after, def->num_components, def->bit_size,
nir_imm_int(after, offset), .align_mul = MIN2(comp_size, stack_alignment));
break;