DAG: Fix assuming f16 is the only 16-bit fp type in concat vector combine (#121637)
[llvm-project.git] / polly / test / Simplify / gemm.ll
blob5120de2db7677ecaa16df6b092677061402e5dce
1 ; RUN: opt %loadNPMPolly '-passes=polly-import-jscop,print<polly-simplify>' -polly-import-jscop-postfix=transformed -disable-output < %s | FileCheck %s
3 ;    void gemm(float A[][1024], float B[][1024], float C[][1024]) {
4 ;      for (long i = 0; i < 1024; i++)
5 ;        for (long j = 0; j < 1024; j++) {
6 ;          float tmp = C[i][j];
7 ;          for (long k = 0; k < 1024; k++)
8 ;            tmp += A[i][k] * B[k][j];
9 ;          C[i][j] = tmp;
10 ;        }
11 ;    }
13 ; CHECK:      After accesses {
14 ; CHECK-NEXT:     Stmt_bb13
15 ; CHECK-NEXT:             MustWriteAccess :=    [Reduction Type: NONE] [Scalar: 1]
16 ; CHECK-NEXT:                 { Stmt_bb13[i0, i1, i2] -> MemRef_tmp_0__phi[] };
17 ; CHECK-NEXT:            new: { Stmt_bb13[i0, i1, i2] -> MemRef_C[i0, i1] };
18 ; CHECK-NEXT:             ReadAccess := [Reduction Type: NONE] [Scalar: 0]
19 ; CHECK-NEXT:                 { Stmt_bb13[i0, i1, i2] -> MemRef_A[i0, i2] };
20 ; CHECK-NEXT:             ReadAccess := [Reduction Type: NONE] [Scalar: 0]
21 ; CHECK-NEXT:                 { Stmt_bb13[i0, i1, i2] -> MemRef_B[i2, i1] };
22 ; CHECK-NEXT:             ReadAccess := [Reduction Type: NONE] [Scalar: 1]
23 ; CHECK-NEXT:                 { Stmt_bb13[i0, i1, i2] -> MemRef_tmp_0[] };
24 ; CHECK-NEXT:            new: { Stmt_bb13[i0, i1, i2] -> MemRef_C[i0, i1] };
25 ; CHECK-NEXT: }
27 target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
28 target triple = "x86_64-pc-linux-gnu"
30 define void @gemm(ptr %A, ptr %B, ptr %C) {
31 bb:
32   br label %bb3
34 bb3:                                              ; preds = %bb26, %bb
35   %i.0 = phi i64 [ 0, %bb ], [ %tmp27, %bb26 ]
36   %exitcond2 = icmp ne i64 %i.0, 1024
37   br i1 %exitcond2, label %bb5, label %bb4
39 bb4:                                              ; preds = %bb3
40   br label %bb28
42 bb5:                                              ; preds = %bb3
43   br label %bb6
45 bb6:                                              ; preds = %bb23, %bb5
46   %j.0 = phi i64 [ 0, %bb5 ], [ %tmp24, %bb23 ]
47   %exitcond1 = icmp ne i64 %j.0, 1024
48   br i1 %exitcond1, label %bb8, label %bb7
50 bb7:                                              ; preds = %bb6
51   br label %bb25
53 bb8:                                              ; preds = %bb6
54   %tmp = getelementptr inbounds [1024 x float], ptr %C, i64 %i.0, i64 %j.0
55   %tmp9 = load float, ptr %tmp, align 4, !tbaa !1
56   br label %bb10
58 bb10:                                             ; preds = %bb13, %bb8
59   %tmp.0 = phi float [ %tmp9, %bb8 ], [ %tmp19, %bb13 ]
60   %k.0 = phi i64 [ 0, %bb8 ], [ %tmp20, %bb13 ]
61   %exitcond = icmp ne i64 %k.0, 1024
62   br i1 %exitcond, label %bb12, label %bb11
64 bb11:                                             ; preds = %bb10
65   %tmp.0.lcssa = phi float [ %tmp.0, %bb10 ]
66   br label %bb21
68 bb12:                                             ; preds = %bb10
69   br label %bb13
71 bb13:                                             ; preds = %bb12
72   %tmp14 = getelementptr inbounds [1024 x float], ptr %A, i64 %i.0, i64 %k.0
73   %tmp15 = load float, ptr %tmp14, align 4, !tbaa !1
74   %tmp16 = getelementptr inbounds [1024 x float], ptr %B, i64 %k.0, i64 %j.0
75   %tmp17 = load float, ptr %tmp16, align 4, !tbaa !1
76   %tmp18 = fmul float %tmp15, %tmp17
77   %tmp19 = fadd float %tmp.0, %tmp18
78   %tmp20 = add nuw nsw i64 %k.0, 1
79   br label %bb10
81 bb21:                                             ; preds = %bb11
82   %tmp22 = getelementptr inbounds [1024 x float], ptr %C, i64 %i.0, i64 %j.0
83   store float %tmp.0.lcssa, ptr %tmp22, align 4, !tbaa !1
84   br label %bb23
86 bb23:                                             ; preds = %bb21
87   %tmp24 = add nuw nsw i64 %j.0, 1
88   br label %bb6
90 bb25:                                             ; preds = %bb7
91   br label %bb26
93 bb26:                                             ; preds = %bb25
94   %tmp27 = add nuw nsw i64 %i.0, 1
95   br label %bb3
97 bb28:                                             ; preds = %bb4
98   ret void
101 declare void @llvm.lifetime.start(i64, ptr nocapture)
103 declare void @llvm.lifetime.end(i64, ptr nocapture)
106 !llvm.ident = !{!0}
108 !0 = !{!"Ubuntu clang version 3.7.1-3ubuntu4 (tags/RELEASE_371/final) (based on LLVM 3.7.1)"}
109 !1 = !{!2, !2, i64 0}
110 !2 = !{!"float", !3, i64 0}
111 !3 = !{!"omnipotent char", !4, i64 0}
112 !4 = !{!"Simple C/C++ TBAA"}