dondragmer

#pragma use_dxc //enable SM 6.0 features, in Unity this is only supported on version 2020.2.0a8 or later with D3D12 enabled
#pragma kernel CountTotalsInBlock
#pragma kernel BlockCountPostfixSum
#pragma kernel CalculateOffsetsForEachKey
#pragma kernel FinalSort

uint _FirstBitToSort;
int _NumElements;
int _NumBlocks;
bool _ShouldSortPayload;

dondragmer

Buffer<uint> Input;
RWBuffer<uint> Output;

//returns the index that this value should be moved to to sort the array
uint CuteSort(uint value, uint laneIndex)
{
    uint smallerValuesMask = 0;
    uint equalValuesMask = ~0;
    
    //don't need to test every bit if your value is constrained to a smaller range