xref: /llvm-project/llvm/test/Transforms/InstCombine/X86/x86-amx.ll (revision fcfc31fffb9a83416453e60bd0dff2df93c2ee20)
10875c2f7SLuo, Yuanke; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
20875c2f7SLuo, Yuanke; RUN: opt -passes=instcombine -S < %s | FileCheck %s
30875c2f7SLuo, Yuanke
4*fcfc31ffSNikita Popovdefine void @foo(ptr %arrayidx16, ptr %arrayidx29, ptr %arrayidx35, i1 %c1, i1 %c2) {
50875c2f7SLuo, Yuanke; CHECK-LABEL: @foo(
60875c2f7SLuo, Yuanke; CHECK-NEXT:  entry:
70875c2f7SLuo, Yuanke; CHECK-NEXT:    br label [[FOR_COND9:%.*]]
80875c2f7SLuo, Yuanke; CHECK:       for.cond9:
9952e0693SNuno Lopes; CHECK-NEXT:    br i1 [[C1:%.*]], label [[FOR_BODY14:%.*]], label [[EXIT:%.*]]
100875c2f7SLuo, Yuanke; CHECK:       for.body14:
11*fcfc31ffSNikita Popov; CHECK-NEXT:    [[T5:%.*]] = load <256 x i32>, ptr [[ARRAYIDX16:%.*]], align 64
120875c2f7SLuo, Yuanke; CHECK-NEXT:    br label [[FOR_COND18:%.*]]
130875c2f7SLuo, Yuanke; CHECK:       for.cond18:
1466fbf5faSLuo, Yuanke; CHECK-NEXT:    [[SUB_C_SROA_0_0:%.*]] = phi <256 x i32> [ [[T5]], [[FOR_BODY14]] ], [ [[T12:%.*]], [[FOR_BODY24:%.*]] ]
15952e0693SNuno Lopes; CHECK-NEXT:    br i1 [[C2:%.*]], label [[FOR_BODY24]], label [[FOR_COND_CLEANUP23:%.*]]
160875c2f7SLuo, Yuanke; CHECK:       for.cond.cleanup23:
17*fcfc31ffSNikita Popov; CHECK-NEXT:    store <256 x i32> [[SUB_C_SROA_0_0]], ptr [[ARRAYIDX16]], align 64
180875c2f7SLuo, Yuanke; CHECK-NEXT:    br label [[FOR_COND9]]
190875c2f7SLuo, Yuanke; CHECK:       for.body24:
20*fcfc31ffSNikita Popov; CHECK-NEXT:    [[T6:%.*]] = load <256 x i32>, ptr [[ARRAYIDX29:%.*]], align 64
21*fcfc31ffSNikita Popov; CHECK-NEXT:    [[T7:%.*]] = load <256 x i32>, ptr [[ARRAYIDX35:%.*]], align 64
2266fbf5faSLuo, Yuanke; CHECK-NEXT:    [[T8:%.*]] = bitcast <256 x i32> [[SUB_C_SROA_0_0]] to x86_amx
230875c2f7SLuo, Yuanke; CHECK-NEXT:    [[T9:%.*]] = bitcast <256 x i32> [[T6]] to x86_amx
240875c2f7SLuo, Yuanke; CHECK-NEXT:    [[T10:%.*]] = bitcast <256 x i32> [[T7]] to x86_amx
2566fbf5faSLuo, Yuanke; CHECK-NEXT:    [[T11:%.*]] = call x86_amx @llvm.x86.tdpbssd.internal(i16 1, i16 4, i16 4, x86_amx [[T8]], x86_amx [[T9]], x86_amx [[T10]])
2666fbf5faSLuo, Yuanke; CHECK-NEXT:    [[T12]] = bitcast x86_amx [[T11]] to <256 x i32>
270875c2f7SLuo, Yuanke; CHECK-NEXT:    br label [[FOR_COND18]]
280875c2f7SLuo, Yuanke; CHECK:       exit:
290875c2f7SLuo, Yuanke; CHECK-NEXT:    ret void
300875c2f7SLuo, Yuanke;
310875c2f7SLuo, Yuankeentry:
320875c2f7SLuo, Yuanke  br label %for.cond9
330875c2f7SLuo, Yuankefor.cond9:                                        ; preds = %for.cond, %for.cond.cleanup23
34952e0693SNuno Lopes  br i1 %c1, label %for.body14, label %exit
350875c2f7SLuo, Yuanke
360875c2f7SLuo, Yuankefor.body14:
37*fcfc31ffSNikita Popov  %t5 = load <256 x i32>, ptr %arrayidx16, align 64
380875c2f7SLuo, Yuanke  br label %for.cond18
390875c2f7SLuo, Yuanke
400875c2f7SLuo, Yuankefor.cond18:                                       ; preds = %for.body24, %for.body14
410875c2f7SLuo, Yuanke  %sub_c.sroa.0.0 = phi <256 x i32> [ %t5, %for.body14 ], [ %t12, %for.body24 ]
42952e0693SNuno Lopes  br i1 %c2, label %for.body24, label %for.cond.cleanup23
430875c2f7SLuo, Yuanke
440875c2f7SLuo, Yuankefor.cond.cleanup23:                               ; preds = %for.cond18
45*fcfc31ffSNikita Popov  store <256 x i32> %sub_c.sroa.0.0, ptr %arrayidx16, align 64
460875c2f7SLuo, Yuanke  br label %for.cond9
470875c2f7SLuo, Yuanke
480875c2f7SLuo, Yuankefor.body24:                                       ; preds = %for.cond18
49*fcfc31ffSNikita Popov  %t6 = load <256 x i32>, ptr %arrayidx29, align 64
50*fcfc31ffSNikita Popov  %t7 = load <256 x i32>, ptr %arrayidx35, align 64
510875c2f7SLuo, Yuanke  %t8 = bitcast <256 x i32> %sub_c.sroa.0.0 to x86_amx
520875c2f7SLuo, Yuanke  %t9 = bitcast <256 x i32> %t6 to x86_amx
530875c2f7SLuo, Yuanke  %t10 = bitcast <256 x i32> %t7 to x86_amx
540875c2f7SLuo, Yuanke  %t11 = call x86_amx @llvm.x86.tdpbssd.internal(i16 1, i16 4, i16 4, x86_amx %t8, x86_amx %t9, x86_amx %t10) #12
550875c2f7SLuo, Yuanke  %t12 = bitcast x86_amx %t11 to <256 x i32>
560875c2f7SLuo, Yuanke  br label %for.cond18
570875c2f7SLuo, Yuanke
580875c2f7SLuo, Yuankeexit:
590875c2f7SLuo, Yuanke  ret void
600875c2f7SLuo, Yuanke}
610875c2f7SLuo, Yuanke
62*fcfc31ffSNikita Popovdeclare x86_amx @llvm.x86.tileloadd64.internal(i16, i16, ptr, i64)
630875c2f7SLuo, Yuankedeclare x86_amx @llvm.x86.tdpbssd.internal(i16, i16, i16, x86_amx, x86_amx, x86_amx)
64*fcfc31ffSNikita Popovdeclare void @llvm.x86.tilestored64.internal(i16, i16, ptr, i64, x86_amx)
65