@@ -227,30 +227,6 @@ func.func @nvvm_mma_m16n8k16_f16_f16(%a0 : vector<2xf16>, %a1 : vector<2xf16>,
227227 llvm.return %0 : !llvm.struct <(vector <2 xf16 >, vector <2 xf16 >)>
228228}
229229
230- // CHECK-LABEL: @nvvm_mma_m16n8k16_f32_f16
231- func.func @nvvm_mma_m16n8k16_f32_f16 (%a0 : vector <2 xf16 >, %a1 : vector <2 xf16 >,
232- %a2 : vector <2 xf16 >, %a3 : vector <2 xf16 >,
233- %b0 : vector <2 xf16 >, %b1 : vector <2 xf16 >,
234- %c0 : vector <2 xf16 >, %c1 : vector <2 xf16 >) {
235- // CHECK: nvvm.mma.sync A[{{.*}}, {{.*}}, {{.*}}, {{.*}}] B[{{.*}}, {{.*}}] C[{{.*}}, {{.*}}] {layoutA = #nvvm.mma_layout<row>, layoutB = #nvvm.mma_layout<col>, shape = #nvvm.shape<m = 16, n = 8, k = 16>} : (vector<2xf16>, vector<2xf16>, vector<2xf16>) -> !llvm.struct<(f32, f32, f32, f32)>
236- %0 = nvvm.mma.sync A [%a0 , %a1 , %a2 , %a3 ] B [%b0 , %b1 ] C [%c0 , %c1 ]
237- {layoutA = #nvvm.mma_layout <row >, layoutB = #nvvm.mma_layout <col >,
238- shape = #nvvm.shape <m = 16 , n = 8 , k = 16 >} : (vector <2 xf16 >,vector <2 xf16 >,vector <2 xf16 >) -> !llvm.struct <(f32 , f32 , f32 , f32 )>
239- llvm.return %0 : !llvm.struct <(f32 , f32 , f32 , f32 )>
240- }
241-
242- // CHECK-LABEL: @nvvm_mma_m16n8k16_f16_f32
243- func.func @nvvm_mma_m16n8k16_f16_f32 (%a0 : vector <2 xf16 >, %a1 : vector <2 xf16 >,
244- %a2 : vector <2 xf16 >, %a3 : vector <2 xf16 >,
245- %b0 : vector <2 xf16 >, %b1 : vector <2 xf16 >,
246- %c0 : f32 , %c1 : f32 , %c2 : f32 , %c3 : f32 ) {
247- // CHECK: nvvm.mma.sync A[{{.*}}, {{.*}}, {{.*}}, {{.*}}] B[{{.*}}, {{.*}}] C[{{.*}}, {{.*}}, {{.*}}, {{.*}}] {layoutA = #nvvm.mma_layout<row>, layoutB = #nvvm.mma_layout<col>, shape = #nvvm.shape<m = 16, n = 8, k = 16>} : (vector<2xf16>, vector<2xf16>, f32) -> !llvm.struct<(vector<2xf16>, vector<2xf16>)>
248- %0 = nvvm.mma.sync A [%a0 , %a1 , %a2 , %a3 ] B [%b0 , %b1 ] C [%c0 , %c1 , %c2 , %c3 ]
249- {layoutA = #nvvm.mma_layout <row >, layoutB = #nvvm.mma_layout <col >,
250- shape = #nvvm.shape <m = 16 , n = 8 , k = 16 >} : (vector <2 xf16 >, vector <2 xf16 >, f32 ) -> !llvm.struct <(vector <2 xf16 >, vector <2 xf16 >)>
251- llvm.return %0 : !llvm.struct <(vector <2 xf16 >, vector <2 xf16 >)>
252- }
253-
254230// CHECK-LABEL: @nvvm_mma_m16n8k16_f32_f32
255231func.func @nvvm_mma_m16n8k16_f32_f32 (%a0 : vector <2 xf16 >, %a1 : vector <2 xf16 >,
256232 %a2 : vector <2 xf16 >, %a3 : vector <2 xf16 >,
0 commit comments