mirror of
https://github.com/gfx-rs/wgpu.git
synced 2025-12-08 21:26:17 +00:00
9c023e5e29
* Rudimentary impl of quad ops, impl quad ops for spirv * Impl quad swap for hlsl, msl and wgsl, finish spv front * Cargo clippy & cargo fmt, impl valid for quad ops * Enable quad feature * Add missing feature to glsl * Simplifying code by making `SubgroupQuadSwap` an instance of `SubgroupGather` * Add `GroupNonUniformQuad` spv capability to Vulkan * Adding GPU tests for quad operations * Validate that broadcast operations use const invocation ids * Added changelog entry --------- Co-authored-by: valaphee <32491319+valaphee@users.noreply.github.com>
43 lines
2.2 KiB
HLSL
43 lines
2.2 KiB
HLSL
struct Structure {
|
|
uint num_subgroups;
|
|
uint subgroup_size;
|
|
};
|
|
|
|
struct ComputeInput_main {
|
|
uint __local_invocation_index : SV_GroupIndex;
|
|
};
|
|
|
|
[numthreads(1, 1, 1)]
|
|
void main(ComputeInput_main computeinput_main)
|
|
{
|
|
Structure sizes = { (1u + WaveGetLaneCount() - 1u) / WaveGetLaneCount(), WaveGetLaneCount() };
|
|
uint subgroup_id = computeinput_main.__local_invocation_index / WaveGetLaneCount();
|
|
uint subgroup_invocation_id = WaveGetLaneIndex();
|
|
const uint4 _e7 = WaveActiveBallot(((subgroup_invocation_id & 1u) == 1u));
|
|
const uint4 _e8 = WaveActiveBallot(true);
|
|
const bool _e11 = WaveActiveAllTrue((subgroup_invocation_id != 0u));
|
|
const bool _e14 = WaveActiveAnyTrue((subgroup_invocation_id == 0u));
|
|
const uint _e15 = WaveActiveSum(subgroup_invocation_id);
|
|
const uint _e16 = WaveActiveProduct(subgroup_invocation_id);
|
|
const uint _e17 = WaveActiveMin(subgroup_invocation_id);
|
|
const uint _e18 = WaveActiveMax(subgroup_invocation_id);
|
|
const uint _e19 = WaveActiveBitAnd(subgroup_invocation_id);
|
|
const uint _e20 = WaveActiveBitOr(subgroup_invocation_id);
|
|
const uint _e21 = WaveActiveBitXor(subgroup_invocation_id);
|
|
const uint _e22 = WavePrefixSum(subgroup_invocation_id);
|
|
const uint _e23 = WavePrefixProduct(subgroup_invocation_id);
|
|
const uint _e24 = subgroup_invocation_id + WavePrefixSum(subgroup_invocation_id);
|
|
const uint _e25 = subgroup_invocation_id * WavePrefixProduct(subgroup_invocation_id);
|
|
const uint _e26 = WaveReadLaneFirst(subgroup_invocation_id);
|
|
const uint _e28 = WaveReadLaneAt(subgroup_invocation_id, 4u);
|
|
const uint _e33 = WaveReadLaneAt(subgroup_invocation_id, ((sizes.subgroup_size - 1u) - subgroup_invocation_id));
|
|
const uint _e35 = WaveReadLaneAt(subgroup_invocation_id, WaveGetLaneIndex() + 1u);
|
|
const uint _e37 = WaveReadLaneAt(subgroup_invocation_id, WaveGetLaneIndex() - 1u);
|
|
const uint _e41 = WaveReadLaneAt(subgroup_invocation_id, WaveGetLaneIndex() ^ (sizes.subgroup_size - 1u));
|
|
const uint _e43 = QuadReadLaneAt(subgroup_invocation_id, 4u);
|
|
const uint _e44 = QuadReadAcrossX(subgroup_invocation_id);
|
|
const uint _e45 = QuadReadAcrossY(subgroup_invocation_id);
|
|
const uint _e46 = QuadReadAcrossDiagonal(subgroup_invocation_id);
|
|
return;
|
|
}
|