2021-01-27 16:59:47 -07:00
|
|
|
float Helper_ShuffleDown(float x, uint index, uint mask, out bool valid)
|
2019-10-30 21:29:22 -06:00
|
|
|
{
|
|
|
|
uint clamp = mask & 0x1fu;
|
|
|
|
uint segMask = (mask >> 8) & 0x1fu;
|
2021-09-19 06:38:39 -06:00
|
|
|
uint minThreadId = $SUBGROUP_INVOCATION$ & segMask;
|
2019-10-30 21:29:22 -06:00
|
|
|
uint maxThreadId = minThreadId | (clamp & ~segMask);
|
2021-09-19 06:38:39 -06:00
|
|
|
uint srcThreadId = $SUBGROUP_INVOCATION$ + index;
|
2021-01-27 16:59:47 -07:00
|
|
|
valid = srcThreadId <= maxThreadId;
|
2021-09-19 06:38:39 -06:00
|
|
|
float v = $SUBGROUP_BROADCAST$(x, srcThreadId);
|
2021-08-30 11:02:40 -06:00
|
|
|
return valid ? v : x;
|
2019-10-30 21:29:22 -06:00
|
|
|
}
|