fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2025-12-27 06:10:13 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

nir/lower_system_values: Simplify the computation of LocalInvocationIndex

Browse source
This commit is contained in:
Jason Ekstrand 2016-04-10 23:43:34 -07:00
parent bff7a8c4f3
commit 3aa1a5ee88
1 changed files with 8 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
src/compiler/nir/nir_lower_system_values.c
View file

@ -83,7 +83,7 @@ convert_block(nir_block *block, void *void_state)
case SYSTEM_VALUE_LOCAL_INVOCATION_INDEX: {
/* From the GLSL man page for gl_LocalInvocationIndex:
*
* ?The value of gl_LocalInvocationIndex is equal to
* "The value of gl_LocalInvocationIndex is equal to
* gl_LocalInvocationID.z * gl_WorkGroupSize.x *
* gl_WorkGroupSize.y + gl_LocalInvocationID.y *
* gl_WorkGroupSize.x + gl_LocalInvocationID.x"
@ -91,15 +91,14 @@ convert_block(nir_block *block, void *void_state)
nir_ssa_def *local_id =
nir_load_system_value(b, nir_intrinsic_load_local_invocation_id, 0);
unsigned stride_y = b->shader->info.cs.local_size[0];
unsigned stride_z = b->shader->info.cs.local_size[0] *
b->shader->info.cs.local_size[1];
nir_ssa_def *size_x = nir_imm_int(b, b->shader->info.cs.local_size[0]);
nir_ssa_def *size_y = nir_imm_int(b, b->shader->info.cs.local_size[1]);
sysval = nir_iadd(b, nir_imul(b, nir_channel(b, local_id, 2),
nir_imm_int(b, stride_z)),
nir_iadd(b, nir_imul(b, nir_channel(b, local_id, 1),
nir_imm_int(b, stride_y)),
nir_channel(b, local_id, 0)));
sysval = nir_imul(b, nir_channel(b, local_id, 2),
nir_imul(b, size_x, size_y));
sysval = nir_iadd(b, sysval,
nir_imul(b, nir_channel(b, local_id, 1), size_x));
sysval = nir_iadd(b, sysval, nir_channel(b, local_id, 0));
break;
}