blob: bc09bbc6e6e1b075381d00f7dca3ea5071ff9409 [file] [log] [blame]
; SPIR-V
; Version: 1.3
; Generator: Google Tint Compiler; 1
; Bound: 46
; Schema: 0
OpCapability Shader
OpCapability VulkanMemoryModel
OpCapability VulkanMemoryModelDeviceScope
OpCapability CooperativeMatrixKHR
OpCapability Float16
OpCapability UniformAndStorageBuffer16BitAccess
OpCapability StorageBuffer16BitAccess
OpExtension "SPV_KHR_vulkan_memory_model"
OpExtension "SPV_KHR_cooperative_matrix"
OpMemoryModel Logical Vulkan
OpEntryPoint GLCompute %compute_main "compute_main"
OpExecutionMode %compute_main LocalSize 1 1 1
OpMemberName %prevent_dce_block 0 "inner"
OpName %prevent_dce_block "prevent_dce_block"
OpName %subgroupMatrixMultiplyAccumulate_071472 "subgroupMatrixMultiplyAccumulate_071472"
OpName %arg_0 "arg_0"
OpName %arg_1 "arg_1"
OpName %arg_2 "arg_2"
OpName %res "res"
OpName %compute_main "compute_main"
OpDecorate %_arr_float_uint_1024 ArrayStride 4
OpMemberDecorate %prevent_dce_block 0 Offset 0
OpDecorate %prevent_dce_block Block
OpDecorate %1 DescriptorSet 0
OpDecorate %1 Binding 0
%float = OpTypeFloat 32
%uint = OpTypeInt 32 0
%uint_1024 = OpConstant %uint 1024
%_arr_float_uint_1024 = OpTypeArray %float %uint_1024
%prevent_dce_block = OpTypeStruct %_arr_float_uint_1024
%_ptr_StorageBuffer_prevent_dce_block = OpTypePointer StorageBuffer %prevent_dce_block
%1 = OpVariable %_ptr_StorageBuffer_prevent_dce_block StorageBuffer
%uint_3 = OpConstant %uint 3
%uint_8 = OpConstant %uint 8
%uint_2 = OpConstant %uint 2
%9 = OpTypeCooperativeMatrixKHR %float %uint_3 %uint_8 %uint_8 %uint_2
%13 = OpTypeFunction %9
%half = OpTypeFloat 16
%uint_0 = OpConstant %uint 0
%16 = OpTypeCooperativeMatrixKHR %half %uint_3 %uint_8 %uint_8 %uint_0
%15 = OpConstantNull %16
%_ptr_Function_16 = OpTypePointer Function %16
%uint_1 = OpConstant %uint 1
%22 = OpTypeCooperativeMatrixKHR %half %uint_3 %uint_8 %uint_8 %uint_1
%21 = OpConstantNull %22
%_ptr_Function_22 = OpTypePointer Function %22
%26 = OpConstantNull %9
%_ptr_Function_9 = OpTypePointer Function %9
%void = OpTypeVoid
%37 = OpTypeFunction %void
%_ptr_StorageBuffer__arr_float_uint_1024 = OpTypePointer StorageBuffer %_arr_float_uint_1024
%_ptr_StorageBuffer_float = OpTypePointer StorageBuffer %float
%uint_64 = OpConstant %uint 64
%subgroupMatrixMultiplyAccumulate_071472 = OpFunction %9 None %13
%14 = OpLabel
%arg_0 = OpVariable %_ptr_Function_16 Function
%arg_1 = OpVariable %_ptr_Function_22 Function
%arg_2 = OpVariable %_ptr_Function_9 Function
%res = OpVariable %_ptr_Function_9 Function
OpStore %arg_0 %15
OpStore %arg_1 %21
OpStore %arg_2 %26
%29 = OpLoad %16 %arg_0 None
%30 = OpLoad %22 %arg_1 None
%31 = OpLoad %9 %arg_2 None
%32 = OpCooperativeMatrixMulAddKHR %9 %29 %30 %31 NoneKHR
OpStore %res %32
%34 = OpLoad %9 %res None
OpReturnValue %34
OpFunctionEnd
%compute_main = OpFunction %void None %37
%38 = OpLabel
%39 = OpFunctionCall %9 %subgroupMatrixMultiplyAccumulate_071472
%40 = OpAccessChain %_ptr_StorageBuffer__arr_float_uint_1024 %1 %uint_0
%42 = OpAccessChain %_ptr_StorageBuffer_float %40 %uint_0
OpCooperativeMatrixStoreKHR %42 %39 %uint_0 %uint_64 NonPrivatePointer
OpReturn
OpFunctionEnd