blob: b85404bc217804ef675a09ee6f4d8089c81392a1 [file] [log] [blame]
; SPIR-V
; Version: 1.3
; Generator: Google Tint Compiler; 1
; Bound: 75
; Schema: 0
OpCapability Shader
OpCapability VulkanMemoryModel
OpCapability VulkanMemoryModelDeviceScope
OpCapability CooperativeMatrixKHR
OpExtension "SPV_KHR_vulkan_memory_model"
OpExtension "SPV_KHR_cooperative_matrix"
%26 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical Vulkan
OpEntryPoint GLCompute %compute_main "compute_main" %compute_main_local_invocation_index_Input
OpExecutionMode %compute_main LocalSize 1 1 1
OpMemberName %prevent_dce_block_tint_explicit_layout 0 "inner"
OpName %prevent_dce_block_tint_explicit_layout "prevent_dce_block_tint_explicit_layout"
OpName %arg_0 "arg_0"
OpName %compute_main_local_invocation_index_Input "compute_main_local_invocation_index_Input"
OpName %subgroupMatrixLoad_d6e273 "subgroupMatrixLoad_d6e273"
OpName %arg_1 "arg_1"
OpName %arg_3 "arg_3"
OpName %res "res"
OpName %compute_main_inner "compute_main_inner"
OpName %tint_local_index "tint_local_index"
OpName %compute_main "compute_main"
OpDecorate %_arr_uint_uint_1024 ArrayStride 4
OpMemberDecorate %prevent_dce_block_tint_explicit_layout 0 Offset 0
OpDecorate %prevent_dce_block_tint_explicit_layout Block
OpDecorate %1 DescriptorSet 0
OpDecorate %1 Binding 0
OpDecorate %compute_main_local_invocation_index_Input BuiltIn LocalInvocationIndex
%uint = OpTypeInt 32 0
%uint_1024 = OpConstant %uint 1024
%_arr_uint_uint_1024 = OpTypeArray %uint %uint_1024
%prevent_dce_block_tint_explicit_layout = OpTypeStruct %_arr_uint_uint_1024
%_ptr_StorageBuffer_prevent_dce_block_tint_explicit_layout = OpTypePointer StorageBuffer %prevent_dce_block_tint_explicit_layout
%1 = OpVariable %_ptr_StorageBuffer_prevent_dce_block_tint_explicit_layout StorageBuffer
%_arr_uint_uint_1024_0 = OpTypeArray %uint %uint_1024
%_ptr_Workgroup__arr_uint_uint_1024_0 = OpTypePointer Workgroup %_arr_uint_uint_1024_0
%arg_0 = OpVariable %_ptr_Workgroup__arr_uint_uint_1024_0 Workgroup
%_ptr_Input_uint = OpTypePointer Input %uint
%compute_main_local_invocation_index_Input = OpVariable %_ptr_Input_uint Input
%uint_3 = OpConstant %uint 3
%uint_8 = OpConstant %uint 8
%uint_2 = OpConstant %uint 2
%13 = OpTypeCooperativeMatrixKHR %uint %uint_3 %uint_8 %uint_8 %uint_2
%17 = OpTypeFunction %13
%_ptr_Function_uint = OpTypePointer Function %uint
%uint_1 = OpConstant %uint 1
%uint_7 = OpConstant %uint 7
%bool = OpTypeBool
%_ptr_Function_13 = OpTypePointer Function %13
%uint_0 = OpConstant %uint 0
%35 = OpConstantComposite %13 %uint_0
%_ptr_Workgroup_uint = OpTypePointer Workgroup %uint
%void = OpTypeVoid
%48 = OpTypeFunction %void %uint
%uint_24840 = OpConstant %uint 24840
%_ptr_StorageBuffer__arr_uint_uint_1024 = OpTypePointer StorageBuffer %_arr_uint_uint_1024
%_ptr_StorageBuffer_uint = OpTypePointer StorageBuffer %uint
%uint_64 = OpConstant %uint 64
%71 = OpTypeFunction %void
%subgroupMatrixLoad_d6e273 = OpFunction %13 None %17
%18 = OpLabel
%arg_1 = OpVariable %_ptr_Function_uint Function
%arg_3 = OpVariable %_ptr_Function_uint Function
%33 = OpVariable %_ptr_Function_13 Function %35
%res = OpVariable %_ptr_Function_13 Function
OpStore %arg_1 %uint_1
OpStore %arg_3 %uint_8
%23 = OpLoad %uint %arg_1 None
%24 = OpLoad %uint %arg_3 None
%25 = OpExtInst %uint %26 UMax %24 %uint_8
%27 = OpIMul %uint %25 %uint_7
%29 = OpIAdd %uint %23 %27
%30 = OpIAdd %uint %29 %uint_8
%31 = OpULessThanEqual %bool %30 %uint_1024
OpSelectionMerge %37 None
OpBranchConditional %31 %38 %37
%38 = OpLabel
%42 = OpAccessChain %_ptr_Workgroup_uint %arg_0 %23
%44 = OpCooperativeMatrixLoadKHR %13 %42 %uint_1 %25 NonPrivatePointer
OpStore %33 %44 None
OpBranch %37
%37 = OpLabel
%39 = OpLoad %13 %33 None
OpStore %res %39
%41 = OpLoad %13 %res None
OpReturnValue %41
OpFunctionEnd
%compute_main_inner = OpFunction %void None %48
%tint_local_index = OpFunctionParameter %uint
%49 = OpLabel
OpBranch %50
%50 = OpLabel
OpBranch %53
%53 = OpLabel
%55 = OpPhi %uint %tint_local_index %50 %56 %52
OpLoopMerge %54 %52 None
OpBranch %51
%51 = OpLabel
%66 = OpUGreaterThanEqual %bool %55 %uint_1024
OpSelectionMerge %67 None
OpBranchConditional %66 %68 %67
%68 = OpLabel
OpBranch %54
%67 = OpLabel
%69 = OpAccessChain %_ptr_Workgroup_uint %arg_0 %55
OpStore %69 %uint_0 NonPrivatePointer
OpBranch %52
%52 = OpLabel
%56 = OpIAdd %uint %55 %uint_1
OpBranch %53
%54 = OpLabel
OpControlBarrier %uint_2 %uint_2 %uint_24840
%59 = OpFunctionCall %13 %subgroupMatrixLoad_d6e273
%60 = OpAccessChain %_ptr_StorageBuffer__arr_uint_uint_1024 %1 %uint_0
%62 = OpAccessChain %_ptr_StorageBuffer_uint %60 %uint_0
OpCooperativeMatrixStoreKHR %62 %59 %uint_0 %uint_64 NonPrivatePointer
OpReturn
OpFunctionEnd
%compute_main = OpFunction %void None %71
%72 = OpLabel
%73 = OpLoad %uint %compute_main_local_invocation_index_Input None
%74 = OpFunctionCall %void %compute_main_inner %73
OpReturn
OpFunctionEnd