fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2026-01-02 13:50:09 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

r600/sfn: Fix atomic lowering

Browse source 
Fixes: 56dedf052f
  r600/sfn: add r600 specific lowering pass for atomics and use it

Signed-off-by: Gert Wollny <gert.wollny@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/21684>
This commit is contained in:
Gert Wollny 2023-03-03 12:16:40 +01:00 • committed by Marge Bot
parent 3c3ecdab36
commit 9b09f244f0
1 changed files with 22 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
src/gallium/drivers/r600/sfn/sfn_nir.cpp
View file

@ -444,25 +444,32 @@ r600_lower_clipvertex_to_clipdist(nir_shader *sh, pipe_stream_output_info& so_in
static bool
r600_nir_lower_atomics(nir_shader *shader)
{
/* First re-do the offsets, in Hardware we start at zero for each new
* binding, and we use an offset of one per counter */
int current_binding = -1;
int current_offset = 0;
nir_foreach_variable_with_modes(var, shader, nir_var_uniform)
{
if (!var->type->contains_atomic())
continue;
/* In Hardware we start at a zero index for each new
* binding, and we use an offset of one per counter. We also
* need to sort the atomics according to binding and offset. */
std::map<unsigned, unsigned> binding_offset;
std::map<unsigned, nir_variable *> sorted_var;
if (current_binding == (int)var->data.binding) {
var->data.index = current_offset;
current_offset += var->type->atomic_size() / ATOMIC_COUNTER_SIZE;
} else {
current_binding = var->data.binding;
var->data.index = 0;
current_offset = var->type->atomic_size() / ATOMIC_COUNTER_SIZE;
nir_foreach_variable_with_modes_safe(var, shader, nir_var_uniform) {
if (var->type->contains_atomic()) {
sorted_var[(var->data.binding << 16) | var->data.offset] = var;
exec_node_remove(&var->node);
}
}
for (auto& [dummy, var] : sorted_var) {
auto iindex = binding_offset.find(var->data.binding);
unsigned offset_update = var->type->atomic_size() / ATOMIC_COUNTER_SIZE;
if (iindex == binding_offset.end()) {
var->data.index = 0;
binding_offset[var->data.binding] = offset_update;
} else {
var->data.index = iindex->second;
iindex->second += offset_update;
}
shader->variables.push_tail(&var->node);
}
return nir_shader_instructions_pass(shader,
r600_lower_deref_instr,
nir_metadata_block_index | nir_metadata_dominance,