spv.coopmat2_tensor.comp // Module Version 10600 // Generated by (magic number): 8000b // Id's are bound by 170 Capability Shader Capability Float16 Capability StorageUniformBufferBlock16 Capability VulkanMemoryModelKHR Capability PhysicalStorageBufferAddressesEXT Capability CooperativeMatrixTensorAddressingNV Capability CooperativeMatrixBlockLoadsNV Capability TensorAddressingNV Capability CooperativeMatrixKHR Extension "SPV_KHR_cooperative_matrix" Extension "SPV_NV_cooperative_matrix2" Extension "SPV_NV_tensor_addressing" 1: ExtInstImport "GLSL.std.450" MemoryModel PhysicalStorageBuffer64EXT VulkanKHR EntryPoint GLCompute 4 "main" 90 122 ExecutionMode 4 LocalSize 64 1 1 Source GLSL 450 SourceExtension "GL_EXT_buffer_reference" SourceExtension "GL_EXT_shader_explicit_arithmetic_types" SourceExtension "GL_KHR_cooperative_matrix" SourceExtension "GL_KHR_memory_scope_semantics" SourceExtension "GL_NV_cooperative_matrix2" Name 4 "main" Name 8 "fp16Buf" MemberName 8(fp16Buf) 0 "f" Name 16 "decode(bref-fp16Buf1;u1[2];u1[2];" Name 13 "b" Name 14 "blockCoords" Name 15 "coordInBlock" Name 28 "t" Name 35 "t2" Name 56 "v" Name 75 "v2" Name 79 "v3" Name 84 "A" Name 88 "BufType" MemberName 88(BufType) 0 "arr" MemberName 88(BufType) 1 "x" Name 90 "Buf" Name 118 "S" MemberName 118(S) 0 "x" Name 120 "SBuf" MemberName 120(SBuf) 0 "s" Name 122 "sbuf" Name 129 "Clamp" Name 132 "tc" Name 136 "Arr" Name 151 "Clamp_signed" Name 154 "tc_s" Name 158 "v4" Name 162 "tc2" Name 166 "v5" Decorate 8(fp16Buf) Block MemberDecorate 8(fp16Buf) 0 Offset 0 Decorate 13(b) Aliased Decorate 85 ArrayStride 2 Decorate 86 ArrayStride 512 Decorate 87 ArrayStride 2 Decorate 88(BufType) Block MemberDecorate 88(BufType) 0 Offset 0 MemberDecorate 88(BufType) 1 Offset 131072 Decorate 90(Buf) Binding 1 Decorate 90(Buf) DescriptorSet 0 MemberDecorate 118(S) 0 Offset 0 Decorate 119 ArrayStride 4 Decorate 120(SBuf) Block MemberDecorate 120(SBuf) 0 Offset 0 Decorate 122(sbuf) Binding 0 Decorate 122(sbuf) DescriptorSet 0 Decorate 129(Clamp) SpecId 0 Decorate 151(Clamp_signed) SpecId 1 2: TypeVoid 3: TypeFunction 2 TypeForwardPointer 6 PhysicalStorageBufferEXT 7: TypeFloat 16 8(fp16Buf): TypeStruct 7(float16_t) 6: TypePointer PhysicalStorageBufferEXT 8(fp16Buf) 9: TypeInt 32 0 10: 9(int) Constant 2 11: TypeArray 9(int) 10 12: TypeFunction 7(float16_t) 6(ptr) 11 11 18: TypeInt 32 1 19: 18(int) Constant 0 20: TypePointer PhysicalStorageBufferEXT 7(float16_t) 25: 9(int) Constant 0 26: TypeTensorLayoutNV 10 25 27: TypePointer Function 26 29: 18(int) Constant 2 31: 9(int) Constant 3 32: 9(int) Constant 1 33: TypeTensorLayoutNV 31 32 34: TypePointer Function 33 36: 18(int) Constant 3 37: 18(int) Constant 1 40: 9(int) Constant 4 41: 9(int) Constant 8 44: 9(int) Constant 256 45: 9(int) Constant 512 48: 9(int) Constant 128 49: 9(int) Constant 32 51: 9(int) Constant 5 52: TypeBool 53: 52(bool) ConstantFalse 54: TypeTensorViewNV 51 53 25 32 10 31 40 55: TypePointer Function 54 57: 18(int) Constant 5 60: 9(int) Constant 10 61: 9(int) Constant 11 62: 9(int) Constant 12 63: 9(int) Constant 13 64: 9(int) Constant 14 67: 9(int) Constant 15 70: 9(int) Constant 16 72: 52(bool) ConstantTrue 73: TypeTensorViewNV 51 72 25 32 10 31 40 74: TypePointer Function 73 77: TypeTensorViewNV 10 72 32 25 78: TypePointer Function 77 81: 9(int) Constant 64 82: TypeCooperativeMatrixKHR 7(float16_t) 10 81 49 25 83: TypePointer Function 82 85: TypeArray 7(float16_t) 44 86: TypeArray 85 44 87: TypeRuntimeArray 7(float16_t) 88(BufType): TypeStruct 86 87 89: TypePointer StorageBuffer 88(BufType) 90(Buf): 89(ptr) Variable StorageBuffer 91: TypePointer StorageBuffer 7(float16_t) 117: TypeVector 7(float16_t) 2 118(S): TypeStruct 117(f16vec2) 119: TypeRuntimeArray 118(S) 120(SBuf): TypeStruct 119 121: TypePointer StorageBuffer 120(SBuf) 122(sbuf): 121(ptr) Variable StorageBuffer 123: TypePointer StorageBuffer 118(S) 129(Clamp): 9(int) SpecConstant 1 130: TypeTensorLayoutNV 10 129(Clamp) 131: TypePointer Function 130 134: TypeArray 82 10 135: TypePointer Function 134 142: 18(int) Constant 8 148: 18(int) Constant 16 151(Clamp_signed): 18(int) SpecConstant 1 152: TypeTensorLayoutNV 10 151(Clamp_signed) 153: TypePointer Function 152 156: TypeTensorViewNV 129(Clamp) 72 25 157: TypePointer Function 156 160: TypeTensorLayoutNV 129(Clamp) 25 161: TypePointer Function 160 164: TypeTensorViewNV 129(Clamp) 53 25 165: TypePointer Function 164 168: TypeVector 9(int) 3 169: 168(ivec3) ConstantComposite 81 32 32 4(main): 2 Function None 3 5: Label 28(t): 27(ptr) Variable Function 35(t2): 34(ptr) Variable Function 56(v): 55(ptr) Variable Function 75(v2): 74(ptr) Variable Function 79(v3): 78(ptr) Variable Function 84(A): 83(ptr) Variable Function 132(tc): 131(ptr) Variable Function 136(Arr): 135(ptr) Variable Function 154(tc_s): 153(ptr) Variable Function 158(v4): 157(ptr) Variable Function 162(tc2): 161(ptr) Variable Function 166(v5): 165(ptr) Variable Function 30: 26 CreateTensorLayoutNV Store 28(t) 30 38: 33 CreateTensorLayoutNV Store 35(t2) 38 39: 26 Load 28(t) 42: 26 TensorLayoutSetBlockSizeNV 39 40 41 Store 28(t) 42 43: 26 Load 28(t) 46: 26 TensorLayoutSetDimensionNV 43 44 45 Store 28(t) 46 47: 26 Load 28(t) 50: 26 TensorLayoutSliceNV 47 48 49 44 49 Store 28(t) 50 58: 54 CreateTensorViewNV Store 56(v) 58 59: 54 Load 56(v) 65: 54 TensorViewSetDimensionNV 59 60 61 62 63 64 Store 56(v) 65 66: 54 Load 56(v) 68: 54 TensorViewSetStrideNV 66 60 61 62 63 67 Store 56(v) 68 69: 54 Load 56(v) 71: 54 TensorViewSetClipNV 69 25 70 25 70 Store 56(v) 71 76: 73 CreateTensorViewNV Store 75(v2) 76 80: 77 CreateTensorViewNV Store 79(v3) 80 92: 91(ptr) AccessChain 90(Buf) 37 25 93: 26 Load 28(t) 94: 82 Load 84(A) 95: 82 CooperativeMatrixLoadTensorNV 92 94 93 None None Store 84(A) 95 96: 82 Load 84(A) 97: 91(ptr) AccessChain 90(Buf) 37 25 98: 26 Load 28(t) CooperativeMatrixStoreTensorNV 97 96 98 None None 99: 91(ptr) AccessChain 90(Buf) 37 25 100: 26 Load 28(t) 101: 54 Load 56(v) 102: 82 Load 84(A) 103: 82 CooperativeMatrixLoadTensorNV 99 102 100 None TensorView 101 Store 84(A) 103 104: 82 Load 84(A) 105: 91(ptr) AccessChain 90(Buf) 37 25 106: 26 Load 28(t) 107: 54 Load 56(v) CooperativeMatrixStoreTensorNV 105 104 106 None TensorView 107 108: 91(ptr) AccessChain 90(Buf) 37 25 109: 26 Load 28(t) 110: 82 Load 84(A) 111: 82 CooperativeMatrixLoadTensorNV 108 110 109 None DecodeFunc 16(decode(bref-fp16Buf1;u1[2];u1[2];) Store 84(A) 111 112: 91(ptr) AccessChain 90(Buf) 37 25 113: 26 Load 28(t) 114: 54 Load 56(v) 115: 82 Load 84(A) 116: 82 CooperativeMatrixLoadTensorNV 112 115 113 None TensorView DecodeFunc 114 16(decode(bref-fp16Buf1;u1[2];u1[2];) Store 84(A) 116 124: 123(ptr) AccessChain 122(sbuf) 19 32 125: 26 Load 28(t) 126: 54 Load 56(v) 127: 82 Load 84(A) 128: 82 CooperativeMatrixLoadTensorNV 124 127 125 None TensorView DecodeFunc 126 16(decode(bref-fp16Buf1;u1[2];u1[2];) Store 84(A) 128 133: 130 CreateTensorLayoutNV Store 132(tc) 133 137: 83(ptr) AccessChain 136(Arr) 37 138: 91(ptr) AccessChain 90(Buf) 37 25 139: 26 Load 28(t) 140: 82 Load 137 141: 82 CooperativeMatrixLoadTensorNV 138 140 139 None None Store 137 141 143: 91(ptr) AccessChain 90(Buf) 19 142 25 144: 26 Load 28(t) 145: 82 Load 84(A) 146: 82 CooperativeMatrixLoadTensorNV 143 145 144 None None Store 84(A) 146 147: 82 Load 84(A) 149: 91(ptr) AccessChain 90(Buf) 19 148 25 150: 26 Load 28(t) CooperativeMatrixStoreTensorNV 149 147 150 None None 155: 152 CreateTensorLayoutNV Store 154(tc_s) 155 159: 156 CreateTensorViewNV Store 158(v4) 159 163: 160 CreateTensorLayoutNV Store 162(tc2) 163 167: 164 CreateTensorViewNV Store 166(v5) 167 Return FunctionEnd 16(decode(bref-fp16Buf1;u1[2];u1[2];):7(float16_t) Function None 12 13(b): 6(ptr) FunctionParameter 14(blockCoords): 11 FunctionParameter 15(coordInBlock): 11 FunctionParameter 17: Label 21: 20(ptr) AccessChain 13(b) 19 22:7(float16_t) Load 21 Aligned 2 ReturnValue 22 FunctionEnd