[tint] Implement subgroupShuffle
Implement subgroupShuffle builtin for WGSL, SPIR-V IR backend (not AST),
MSL AST and IR backends and HLSL AST and IR backends. Subgroup builtins
are not yet implemented for the GLSL backend.
A polyfill is necessary in the SPIR-V IR backend to handle the case
where the lane ID is a signed integer.
Bug: 354738715
Change-Id: I3c29c1862d29600457fdb790b954fe0854008b52
Reviewed-on: https://dawn-review.googlesource.com/c/dawn/+/201956
Reviewed-by: dan sinclair <dsinclair@chromium.org>
Commit-Queue: Natalie Chouinard <chouinard@google.com>
diff --git a/test/tint/builtins/gen/literal/subgroupShuffle/bbb06c.wgsl.expected.ir.spvasm b/test/tint/builtins/gen/literal/subgroupShuffle/bbb06c.wgsl.expected.ir.spvasm
new file mode 100644
index 0000000..e8e9c47
--- /dev/null
+++ b/test/tint/builtins/gen/literal/subgroupShuffle/bbb06c.wgsl.expected.ir.spvasm
@@ -0,0 +1,60 @@
+; SPIR-V
+; Version: 1.3
+; Generator: Google Tint Compiler; 1
+; Bound: 30
+; Schema: 0
+ OpCapability Shader
+ OpCapability GroupNonUniformShuffle
+ OpMemoryModel Logical GLSL450
+ OpEntryPoint Fragment %fragment_main "fragment_main"
+ OpEntryPoint GLCompute %compute_main "compute_main"
+ OpExecutionMode %fragment_main OriginUpperLeft
+ OpExecutionMode %compute_main LocalSize 1 1 1
+ OpMemberName %tint_symbol_1 0 "tint_symbol"
+ OpName %tint_symbol_1 "tint_symbol_1"
+ OpName %subgroupShuffle_bbb06c "subgroupShuffle_bbb06c"
+ OpName %res "res"
+ OpName %fragment_main "fragment_main"
+ OpName %compute_main "compute_main"
+ OpMemberDecorate %tint_symbol_1 0 Offset 0
+ OpDecorate %tint_symbol_1 Block
+ OpDecorate %1 DescriptorSet 0
+ OpDecorate %1 Binding 0
+ %int = OpTypeInt 32 1
+ %v2int = OpTypeVector %int 2
+%tint_symbol_1 = OpTypeStruct %v2int
+%_ptr_StorageBuffer_tint_symbol_1 = OpTypePointer StorageBuffer %tint_symbol_1
+ %1 = OpVariable %_ptr_StorageBuffer_tint_symbol_1 StorageBuffer
+ %7 = OpTypeFunction %v2int
+ %uint = OpTypeInt 32 0
+ %int_1 = OpConstant %int 1
+ %uint_3 = OpConstant %uint 3
+ %14 = OpConstantComposite %v2int %int_1 %int_1
+%_ptr_Function_v2int = OpTypePointer Function %v2int
+ %void = OpTypeVoid
+ %20 = OpTypeFunction %void
+%_ptr_StorageBuffer_v2int = OpTypePointer StorageBuffer %v2int
+ %uint_0 = OpConstant %uint 0
+%subgroupShuffle_bbb06c = OpFunction %v2int None %7
+ %8 = OpLabel
+ %res = OpVariable %_ptr_Function_v2int Function
+ %10 = OpBitcast %uint %int_1
+ %12 = OpGroupNonUniformShuffle %v2int %uint_3 %14 %10
+ OpStore %res %12
+ %17 = OpLoad %v2int %res
+ OpReturnValue %17
+ OpFunctionEnd
+%fragment_main = OpFunction %void None %20
+ %21 = OpLabel
+ %22 = OpFunctionCall %v2int %subgroupShuffle_bbb06c
+ %23 = OpAccessChain %_ptr_StorageBuffer_v2int %1 %uint_0
+ OpStore %23 %22
+ OpReturn
+ OpFunctionEnd
+%compute_main = OpFunction %void None %20
+ %27 = OpLabel
+ %28 = OpFunctionCall %v2int %subgroupShuffle_bbb06c
+ %29 = OpAccessChain %_ptr_StorageBuffer_v2int %1 %uint_0
+ OpStore %29 %28
+ OpReturn
+ OpFunctionEnd
