diff --git a/src/compiler/nir/nir_gather_info.c b/src/compiler/nir/nir_gather_info.c index d71a0f285b4..69a33ade7a8 100644 --- a/src/compiler/nir/nir_gather_info.c +++ b/src/compiler/nir/nir_gather_info.c @@ -211,7 +211,7 @@ mark_whole_variable(nir_shader *shader, nir_variable *var, } const unsigned slots = - var->data.compact ? DIV_ROUND_UP(glsl_get_length(type), 4) + var->data.compact ? DIV_ROUND_UP(var->data.location_frac + glsl_get_length(type), 4) : glsl_count_attribute_slots(type, false); set_io_mask(shader, var, 0, slots, deref, is_output_read); @@ -288,7 +288,7 @@ try_mask_partial_io(nir_shader *shader, nir_variable *var, return false; const unsigned slots = - var->data.compact ? DIV_ROUND_UP(glsl_get_length(type), 4) + var->data.compact ? DIV_ROUND_UP(var->data.location_frac + glsl_get_length(type), 4) : glsl_count_attribute_slots(type, false); if (offset >= slots) { diff --git a/src/freedreno/vulkan/tu_shader.cc b/src/freedreno/vulkan/tu_shader.cc index ffe5d012f8c..3e18722574a 100644 --- a/src/freedreno/vulkan/tu_shader.cc +++ b/src/freedreno/vulkan/tu_shader.cc @@ -924,7 +924,7 @@ tu_gather_xfb_info(nir_shader *nir, struct ir3_stream_output_info *info) nir_foreach_shader_out_variable(var, nir) { unsigned slots = - var->data.compact ? DIV_ROUND_UP(glsl_get_length(var->type), 4) + var->data.compact ? DIV_ROUND_UP(var->data.location_frac + glsl_get_length(var->type), 4) : glsl_count_attribute_slots(var->type, false); for (unsigned i = 0; i < slots; i++) output_map[var->data.location + i] = var->data.driver_location + i; diff --git a/src/gallium/auxiliary/nir/nir_to_tgsi_info.c b/src/gallium/auxiliary/nir/nir_to_tgsi_info.c index 03b19bbc18f..df13d795cc4 100644 --- a/src/gallium/auxiliary/nir/nir_to_tgsi_info.c +++ b/src/gallium/auxiliary/nir/nir_to_tgsi_info.c @@ -335,7 +335,7 @@ void nir_tgsi_scan_shader(const struct nir_shader *nir, type = glsl_get_array_element(type); } - unsigned attrib_count = variable->data.compact ? DIV_ROUND_UP(glsl_get_length(type), 4) : + unsigned attrib_count = variable->data.compact ? DIV_ROUND_UP(variable->data.location_frac + glsl_get_length(type), 4) : glsl_count_attribute_slots(type, nir->info.stage == MESA_SHADER_VERTEX); i = variable->data.driver_location; @@ -435,7 +435,7 @@ void nir_tgsi_scan_shader(const struct nir_shader *nir, type = glsl_get_array_element(type); } - unsigned attrib_count = variable->data.compact ? DIV_ROUND_UP(glsl_get_length(type), 4) : + unsigned attrib_count = variable->data.compact ? DIV_ROUND_UP(variable->data.location_frac + glsl_get_length(type), 4) : glsl_count_attribute_slots(type, false); for (unsigned k = 0; k < attrib_count; k++, i++) { diff --git a/src/gallium/frontends/lavapipe/lvp_pipeline.c b/src/gallium/frontends/lavapipe/lvp_pipeline.c index 71916fd88bf..609a941bad9 100644 --- a/src/gallium/frontends/lavapipe/lvp_pipeline.c +++ b/src/gallium/frontends/lavapipe/lvp_pipeline.c @@ -544,7 +544,7 @@ lvp_shader_xfb_init(struct lvp_shader *shader) memset(output_mapping, 0, sizeof(output_mapping)); nir_foreach_shader_out_variable(var, shader->pipeline_nir->nir) { - unsigned slots = var->data.compact ? DIV_ROUND_UP(glsl_get_length(var->type), 4) + unsigned slots = var->data.compact ? DIV_ROUND_UP(var->data.location_frac + glsl_get_length(var->type), 4) : glsl_count_attribute_slots(var->type, false); for (unsigned i = 0; i < slots; i++) output_mapping[var->data.location + i] = var->data.driver_location + i; diff --git a/src/intel/compiler/brw_fs_nir.cpp b/src/intel/compiler/brw_fs_nir.cpp index 5ee7fe31851..82d0f0a2cee 100644 --- a/src/intel/compiler/brw_fs_nir.cpp +++ b/src/intel/compiler/brw_fs_nir.cpp @@ -72,7 +72,7 @@ fs_visitor::nir_setup_outputs() nir_foreach_shader_out_variable(var, nir) { const int loc = var->data.driver_location; const unsigned var_vec4s = - var->data.compact ? DIV_ROUND_UP(glsl_get_length(var->type), 4) + var->data.compact ? DIV_ROUND_UP(var->data.location_frac + glsl_get_length(var->type), 4) : type_size_vec4(var->type, true); vec4s[loc] = MAX2(vec4s[loc], var_vec4s); }