xref: /aosp_15_r20/external/skia/tests/sksl/compute/Workgroup.asm.comp (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)
1*c8dee2aaSAndroid Build Coastguard Worker               OpCapability Shader
2*c8dee2aaSAndroid Build Coastguard Worker          %1 = OpExtInstImport "GLSL.std.450"
3*c8dee2aaSAndroid Build Coastguard Worker               OpMemoryModel Logical GLSL450
4*c8dee2aaSAndroid Build Coastguard Worker               OpEntryPoint GLCompute %main "main" %sk_GlobalInvocationID
5*c8dee2aaSAndroid Build Coastguard Worker               OpExecutionMode %main LocalSize 256 1 1
6*c8dee2aaSAndroid Build Coastguard Worker               OpName %inputs "inputs"
7*c8dee2aaSAndroid Build Coastguard Worker               OpMemberName %inputs 0 "in_data"
8*c8dee2aaSAndroid Build Coastguard Worker               OpName %outputs "outputs"
9*c8dee2aaSAndroid Build Coastguard Worker               OpMemberName %outputs 0 "out_data"
10*c8dee2aaSAndroid Build Coastguard Worker               OpName %sk_GlobalInvocationID "sk_GlobalInvocationID"
11*c8dee2aaSAndroid Build Coastguard Worker               OpName %shared_data "shared_data"
12*c8dee2aaSAndroid Build Coastguard Worker               OpName %store_vIf "store_vIf"
13*c8dee2aaSAndroid Build Coastguard Worker               OpName %main "main"
14*c8dee2aaSAndroid Build Coastguard Worker               OpName %id "id"
15*c8dee2aaSAndroid Build Coastguard Worker               OpName %rd_id "rd_id"
16*c8dee2aaSAndroid Build Coastguard Worker               OpName %wr_id "wr_id"
17*c8dee2aaSAndroid Build Coastguard Worker               OpName %mask "mask"
18*c8dee2aaSAndroid Build Coastguard Worker               OpName %step "step"
19*c8dee2aaSAndroid Build Coastguard Worker               OpDecorate %_runtimearr_float ArrayStride 4
20*c8dee2aaSAndroid Build Coastguard Worker               OpMemberDecorate %inputs 0 Offset 0
21*c8dee2aaSAndroid Build Coastguard Worker               OpDecorate %inputs BufferBlock
22*c8dee2aaSAndroid Build Coastguard Worker               OpDecorate %4 Binding 0
23*c8dee2aaSAndroid Build Coastguard Worker               OpDecorate %4 DescriptorSet 0
24*c8dee2aaSAndroid Build Coastguard Worker               OpMemberDecorate %outputs 0 Offset 0
25*c8dee2aaSAndroid Build Coastguard Worker               OpDecorate %outputs BufferBlock
26*c8dee2aaSAndroid Build Coastguard Worker               OpDecorate %9 Binding 1
27*c8dee2aaSAndroid Build Coastguard Worker               OpDecorate %9 DescriptorSet 0
28*c8dee2aaSAndroid Build Coastguard Worker               OpDecorate %sk_GlobalInvocationID BuiltIn GlobalInvocationId
29*c8dee2aaSAndroid Build Coastguard Worker               OpDecorate %_arr_float_int_512 ArrayStride 16
30*c8dee2aaSAndroid Build Coastguard Worker      %float = OpTypeFloat 32
31*c8dee2aaSAndroid Build Coastguard Worker%_runtimearr_float = OpTypeRuntimeArray %float
32*c8dee2aaSAndroid Build Coastguard Worker     %inputs = OpTypeStruct %_runtimearr_float
33*c8dee2aaSAndroid Build Coastguard Worker%_ptr_Uniform_inputs = OpTypePointer Uniform %inputs
34*c8dee2aaSAndroid Build Coastguard Worker          %4 = OpVariable %_ptr_Uniform_inputs Uniform
35*c8dee2aaSAndroid Build Coastguard Worker    %outputs = OpTypeStruct %_runtimearr_float
36*c8dee2aaSAndroid Build Coastguard Worker%_ptr_Uniform_outputs = OpTypePointer Uniform %outputs
37*c8dee2aaSAndroid Build Coastguard Worker          %9 = OpVariable %_ptr_Uniform_outputs Uniform
38*c8dee2aaSAndroid Build Coastguard Worker       %uint = OpTypeInt 32 0
39*c8dee2aaSAndroid Build Coastguard Worker     %v3uint = OpTypeVector %uint 3
40*c8dee2aaSAndroid Build Coastguard Worker%_ptr_Input_v3uint = OpTypePointer Input %v3uint
41*c8dee2aaSAndroid Build Coastguard Worker%sk_GlobalInvocationID = OpVariable %_ptr_Input_v3uint Input
42*c8dee2aaSAndroid Build Coastguard Worker        %int = OpTypeInt 32 1
43*c8dee2aaSAndroid Build Coastguard Worker    %int_512 = OpConstant %int 512
44*c8dee2aaSAndroid Build Coastguard Worker%_arr_float_int_512 = OpTypeArray %float %int_512
45*c8dee2aaSAndroid Build Coastguard Worker%_ptr_Workgroup__arr_float_int_512 = OpTypePointer Workgroup %_arr_float_int_512
46*c8dee2aaSAndroid Build Coastguard Worker%shared_data = OpVariable %_ptr_Workgroup__arr_float_int_512 Workgroup
47*c8dee2aaSAndroid Build Coastguard Worker       %void = OpTypeVoid
48*c8dee2aaSAndroid Build Coastguard Worker%_ptr_Function_uint = OpTypePointer Function %uint
49*c8dee2aaSAndroid Build Coastguard Worker%_ptr_Function_float = OpTypePointer Function %float
50*c8dee2aaSAndroid Build Coastguard Worker         %24 = OpTypeFunction %void %_ptr_Function_uint %_ptr_Function_float
51*c8dee2aaSAndroid Build Coastguard Worker%_ptr_Workgroup_float = OpTypePointer Workgroup %float
52*c8dee2aaSAndroid Build Coastguard Worker         %32 = OpTypeFunction %void
53*c8dee2aaSAndroid Build Coastguard Worker      %int_0 = OpConstant %int 0
54*c8dee2aaSAndroid Build Coastguard Worker     %uint_2 = OpConstant %uint 2
55*c8dee2aaSAndroid Build Coastguard Worker%_ptr_Uniform_float = OpTypePointer Uniform %float
56*c8dee2aaSAndroid Build Coastguard Worker     %uint_1 = OpConstant %uint 1
57*c8dee2aaSAndroid Build Coastguard Worker   %uint_264 = OpConstant %uint 264
58*c8dee2aaSAndroid Build Coastguard Worker     %uint_0 = OpConstant %uint 0
59*c8dee2aaSAndroid Build Coastguard Worker     %uint_9 = OpConstant %uint 9
60*c8dee2aaSAndroid Build Coastguard Worker       %bool = OpTypeBool
61*c8dee2aaSAndroid Build Coastguard Worker  %store_vIf = OpFunction %void None %24
62*c8dee2aaSAndroid Build Coastguard Worker         %25 = OpFunctionParameter %_ptr_Function_uint
63*c8dee2aaSAndroid Build Coastguard Worker         %26 = OpFunctionParameter %_ptr_Function_float
64*c8dee2aaSAndroid Build Coastguard Worker         %27 = OpLabel
65*c8dee2aaSAndroid Build Coastguard Worker         %28 = OpLoad %float %26
66*c8dee2aaSAndroid Build Coastguard Worker         %29 = OpLoad %uint %25
67*c8dee2aaSAndroid Build Coastguard Worker         %30 = OpAccessChain %_ptr_Workgroup_float %shared_data %29
68*c8dee2aaSAndroid Build Coastguard Worker               OpStore %30 %28
69*c8dee2aaSAndroid Build Coastguard Worker               OpReturn
70*c8dee2aaSAndroid Build Coastguard Worker               OpFunctionEnd
71*c8dee2aaSAndroid Build Coastguard Worker       %main = OpFunction %void None %32
72*c8dee2aaSAndroid Build Coastguard Worker         %33 = OpLabel
73*c8dee2aaSAndroid Build Coastguard Worker         %id = OpVariable %_ptr_Function_uint Function
74*c8dee2aaSAndroid Build Coastguard Worker      %rd_id = OpVariable %_ptr_Function_uint Function
75*c8dee2aaSAndroid Build Coastguard Worker      %wr_id = OpVariable %_ptr_Function_uint Function
76*c8dee2aaSAndroid Build Coastguard Worker       %mask = OpVariable %_ptr_Function_uint Function
77*c8dee2aaSAndroid Build Coastguard Worker       %step = OpVariable %_ptr_Function_uint Function
78*c8dee2aaSAndroid Build Coastguard Worker         %85 = OpVariable %_ptr_Function_uint Function
79*c8dee2aaSAndroid Build Coastguard Worker         %91 = OpVariable %_ptr_Function_float Function
80*c8dee2aaSAndroid Build Coastguard Worker         %35 = OpLoad %v3uint %sk_GlobalInvocationID
81*c8dee2aaSAndroid Build Coastguard Worker         %36 = OpCompositeExtract %uint %35 0
82*c8dee2aaSAndroid Build Coastguard Worker               OpStore %id %36
83*c8dee2aaSAndroid Build Coastguard Worker         %42 = OpIMul %uint %36 %uint_2
84*c8dee2aaSAndroid Build Coastguard Worker         %43 = OpAccessChain %_ptr_Uniform_float %4 %int_0 %42
85*c8dee2aaSAndroid Build Coastguard Worker         %45 = OpLoad %float %43
86*c8dee2aaSAndroid Build Coastguard Worker         %46 = OpIMul %uint %36 %uint_2
87*c8dee2aaSAndroid Build Coastguard Worker         %47 = OpAccessChain %_ptr_Workgroup_float %shared_data %46
88*c8dee2aaSAndroid Build Coastguard Worker               OpStore %47 %45
89*c8dee2aaSAndroid Build Coastguard Worker         %48 = OpLoad %uint %id
90*c8dee2aaSAndroid Build Coastguard Worker         %49 = OpIMul %uint %48 %uint_2
91*c8dee2aaSAndroid Build Coastguard Worker         %51 = OpIAdd %uint %49 %uint_1
92*c8dee2aaSAndroid Build Coastguard Worker         %52 = OpAccessChain %_ptr_Uniform_float %4 %int_0 %51
93*c8dee2aaSAndroid Build Coastguard Worker         %53 = OpLoad %float %52
94*c8dee2aaSAndroid Build Coastguard Worker         %54 = OpLoad %uint %id
95*c8dee2aaSAndroid Build Coastguard Worker         %55 = OpIMul %uint %54 %uint_2
96*c8dee2aaSAndroid Build Coastguard Worker         %56 = OpIAdd %uint %55 %uint_1
97*c8dee2aaSAndroid Build Coastguard Worker         %57 = OpAccessChain %_ptr_Workgroup_float %shared_data %56
98*c8dee2aaSAndroid Build Coastguard Worker               OpStore %57 %53
99*c8dee2aaSAndroid Build Coastguard Worker               OpControlBarrier %uint_2 %uint_2 %uint_264
100*c8dee2aaSAndroid Build Coastguard Worker               OpStore %step %uint_0
101*c8dee2aaSAndroid Build Coastguard Worker               OpBranch %62
102*c8dee2aaSAndroid Build Coastguard Worker         %62 = OpLabel
103*c8dee2aaSAndroid Build Coastguard Worker               OpLoopMerge %66 %65 None
104*c8dee2aaSAndroid Build Coastguard Worker               OpBranch %63
105*c8dee2aaSAndroid Build Coastguard Worker         %63 = OpLabel
106*c8dee2aaSAndroid Build Coastguard Worker         %67 = OpLoad %uint %step
107*c8dee2aaSAndroid Build Coastguard Worker         %69 = OpULessThan %bool %67 %uint_9
108*c8dee2aaSAndroid Build Coastguard Worker               OpBranchConditional %69 %64 %66
109*c8dee2aaSAndroid Build Coastguard Worker         %64 = OpLabel
110*c8dee2aaSAndroid Build Coastguard Worker         %71 = OpLoad %uint %step
111*c8dee2aaSAndroid Build Coastguard Worker         %72 = OpShiftLeftLogical %uint %uint_1 %71
112*c8dee2aaSAndroid Build Coastguard Worker         %73 = OpISub %uint %72 %uint_1
113*c8dee2aaSAndroid Build Coastguard Worker               OpStore %mask %73
114*c8dee2aaSAndroid Build Coastguard Worker         %74 = OpLoad %uint %id
115*c8dee2aaSAndroid Build Coastguard Worker         %75 = OpLoad %uint %step
116*c8dee2aaSAndroid Build Coastguard Worker         %76 = OpShiftRightLogical %uint %74 %75
117*c8dee2aaSAndroid Build Coastguard Worker         %77 = OpLoad %uint %step
118*c8dee2aaSAndroid Build Coastguard Worker         %78 = OpIAdd %uint %77 %uint_1
119*c8dee2aaSAndroid Build Coastguard Worker         %79 = OpShiftLeftLogical %uint %76 %78
120*c8dee2aaSAndroid Build Coastguard Worker         %80 = OpIAdd %uint %79 %73
121*c8dee2aaSAndroid Build Coastguard Worker               OpStore %rd_id %80
122*c8dee2aaSAndroid Build Coastguard Worker         %81 = OpIAdd %uint %80 %uint_1
123*c8dee2aaSAndroid Build Coastguard Worker         %82 = OpLoad %uint %id
124*c8dee2aaSAndroid Build Coastguard Worker         %83 = OpBitwiseAnd %uint %82 %73
125*c8dee2aaSAndroid Build Coastguard Worker         %84 = OpIAdd %uint %81 %83
126*c8dee2aaSAndroid Build Coastguard Worker               OpStore %wr_id %84
127*c8dee2aaSAndroid Build Coastguard Worker               OpStore %85 %84
128*c8dee2aaSAndroid Build Coastguard Worker         %86 = OpAccessChain %_ptr_Workgroup_float %shared_data %84
129*c8dee2aaSAndroid Build Coastguard Worker         %87 = OpLoad %float %86
130*c8dee2aaSAndroid Build Coastguard Worker         %88 = OpAccessChain %_ptr_Workgroup_float %shared_data %80
131*c8dee2aaSAndroid Build Coastguard Worker         %89 = OpLoad %float %88
132*c8dee2aaSAndroid Build Coastguard Worker         %90 = OpFAdd %float %87 %89
133*c8dee2aaSAndroid Build Coastguard Worker               OpStore %91 %90
134*c8dee2aaSAndroid Build Coastguard Worker         %92 = OpFunctionCall %void %store_vIf %85 %91
135*c8dee2aaSAndroid Build Coastguard Worker               OpControlBarrier %uint_2 %uint_2 %uint_264
136*c8dee2aaSAndroid Build Coastguard Worker               OpBranch %65
137*c8dee2aaSAndroid Build Coastguard Worker         %65 = OpLabel
138*c8dee2aaSAndroid Build Coastguard Worker         %94 = OpLoad %uint %step
139*c8dee2aaSAndroid Build Coastguard Worker         %95 = OpIAdd %uint %94 %uint_1
140*c8dee2aaSAndroid Build Coastguard Worker               OpStore %step %95
141*c8dee2aaSAndroid Build Coastguard Worker               OpBranch %62
142*c8dee2aaSAndroid Build Coastguard Worker         %66 = OpLabel
143*c8dee2aaSAndroid Build Coastguard Worker         %96 = OpLoad %uint %id
144*c8dee2aaSAndroid Build Coastguard Worker         %97 = OpIMul %uint %96 %uint_2
145*c8dee2aaSAndroid Build Coastguard Worker         %98 = OpAccessChain %_ptr_Workgroup_float %shared_data %97
146*c8dee2aaSAndroid Build Coastguard Worker         %99 = OpLoad %float %98
147*c8dee2aaSAndroid Build Coastguard Worker        %100 = OpLoad %uint %id
148*c8dee2aaSAndroid Build Coastguard Worker        %101 = OpIMul %uint %100 %uint_2
149*c8dee2aaSAndroid Build Coastguard Worker        %102 = OpAccessChain %_ptr_Uniform_float %9 %int_0 %101
150*c8dee2aaSAndroid Build Coastguard Worker               OpStore %102 %99
151*c8dee2aaSAndroid Build Coastguard Worker        %103 = OpLoad %uint %id
152*c8dee2aaSAndroid Build Coastguard Worker        %104 = OpIMul %uint %103 %uint_2
153*c8dee2aaSAndroid Build Coastguard Worker        %105 = OpIAdd %uint %104 %uint_1
154*c8dee2aaSAndroid Build Coastguard Worker        %106 = OpAccessChain %_ptr_Workgroup_float %shared_data %105
155*c8dee2aaSAndroid Build Coastguard Worker        %107 = OpLoad %float %106
156*c8dee2aaSAndroid Build Coastguard Worker        %108 = OpLoad %uint %id
157*c8dee2aaSAndroid Build Coastguard Worker        %109 = OpIMul %uint %108 %uint_2
158*c8dee2aaSAndroid Build Coastguard Worker        %110 = OpIAdd %uint %109 %uint_1
159*c8dee2aaSAndroid Build Coastguard Worker        %111 = OpAccessChain %_ptr_Uniform_float %9 %int_0 %110
160*c8dee2aaSAndroid Build Coastguard Worker               OpStore %111 %107
161*c8dee2aaSAndroid Build Coastguard Worker               OpReturn
162*c8dee2aaSAndroid Build Coastguard Worker               OpFunctionEnd
163