blob: c17955fe462f9685c82138fac31d0c3a56ea1a6b [file] [log] [blame]
; SPIR-V
; Version: 1.3
; Generator: Google Tint Compiler; 1
; Bound: 60
; Schema: 0
OpCapability Shader
OpCapability VulkanMemoryModel
OpCapability VulkanMemoryModelDeviceScope
OpCapability CooperativeMatrixKHR
OpExtension "SPV_KHR_vulkan_memory_model"
OpExtension "SPV_KHR_cooperative_matrix"
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_c857d1 "subgroupMatrixLoad_c857d1"
OpName %res "res"
OpName %compute_main_inner "compute_main_inner"
OpName %tint_local_index "tint_local_index"
OpName %compute_main "compute_main"
OpDecorate %_arr_float_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
%float = OpTypeFloat 32
%uint = OpTypeInt 32 0
%uint_1024 = OpConstant %uint 1024
%_arr_float_uint_1024 = OpTypeArray %float %uint_1024
%prevent_dce_block_tint_explicit_layout = OpTypeStruct %_arr_float_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
%uint_64 = OpConstant %uint 64
%_arr_float_uint_64 = OpTypeArray %float %uint_64
%_ptr_Workgroup__arr_float_uint_64 = OpTypePointer Workgroup %_arr_float_uint_64
%arg_0 = OpVariable %_ptr_Workgroup__arr_float_uint_64 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
%15 = OpTypeCooperativeMatrixKHR %float %uint_3 %uint_8 %uint_8 %uint_2
%19 = OpTypeFunction %15
%_ptr_Workgroup_float = OpTypePointer Workgroup %float
%uint_1 = OpConstant %uint 1
%_ptr_Function_15 = OpTypePointer Function %15
%void = OpTypeVoid
%31 = OpTypeFunction %void %uint
%uint_24840 = OpConstant %uint 24840
%_ptr_StorageBuffer__arr_float_uint_1024 = OpTypePointer StorageBuffer %_arr_float_uint_1024
%uint_0 = OpConstant %uint 0
%_ptr_StorageBuffer_float = OpTypePointer StorageBuffer %float
%bool = OpTypeBool
%float_0 = OpConstant %float 0
%56 = OpTypeFunction %void
%subgroupMatrixLoad_c857d1 = OpFunction %15 None %19
%20 = OpLabel
%res = OpVariable %_ptr_Function_15 Function
%21 = OpAccessChain %_ptr_Workgroup_float %arg_0 %uint_1
%24 = OpCooperativeMatrixLoadKHR %15 %21 %uint_1 %uint_1 NonPrivatePointer
OpStore %res %24
%27 = OpLoad %15 %res None
OpReturnValue %27
OpFunctionEnd
%compute_main_inner = OpFunction %void None %31
%tint_local_index = OpFunctionParameter %uint
%32 = OpLabel
OpBranch %33
%33 = OpLabel
OpBranch %36
%36 = OpLabel
%38 = OpPhi %uint %tint_local_index %33 %39 %35
OpLoopMerge %37 %35 None
OpBranch %34
%34 = OpLabel
%49 = OpUGreaterThanEqual %bool %38 %uint_64
OpSelectionMerge %51 None
OpBranchConditional %49 %52 %51
%52 = OpLabel
OpBranch %37
%51 = OpLabel
%53 = OpAccessChain %_ptr_Workgroup_float %arg_0 %38
OpStore %53 %float_0 NonPrivatePointer
OpBranch %35
%35 = OpLabel
%39 = OpIAdd %uint %38 %uint_1
OpBranch %36
%37 = OpLabel
OpControlBarrier %uint_2 %uint_2 %uint_24840
%42 = OpFunctionCall %15 %subgroupMatrixLoad_c857d1
%43 = OpAccessChain %_ptr_StorageBuffer__arr_float_uint_1024 %1 %uint_0
%46 = OpAccessChain %_ptr_StorageBuffer_float %43 %uint_0
OpCooperativeMatrixStoreKHR %46 %42 %uint_0 %uint_64 NonPrivatePointer
OpReturn
OpFunctionEnd
%compute_main = OpFunction %void None %56
%57 = OpLabel
%58 = OpLoad %uint %compute_main_local_invocation_index_Input None
%59 = OpFunctionCall %void %compute_main_inner %58
OpReturn
OpFunctionEnd