llvm-project

1
; RUN: opt %loadNPMPolly -passes=polly-opt-isl -polly-pattern-matching-based-opts=true \
2
; RUN: -debug-only=polly-opt-isl -disable-output \
3
; RUN: -polly-tc-opt=true < %s 2>&1 | FileCheck %s
4
; REQUIRES: asserts
5
;
6
;    for (i = 0; i < _PB_NI; i++)
7
;      for (j = 0; j < _PB_NJ; j++)
8
;      {
9
;        for (k = 0; k < _PB_NK; k++)
10
;        {
11
;          double Mul = A[i][k] * B[k][j];
12
;          D[i][j][k] += Mul;
13
;          C[i][j] += Mul;
14
;        }
15
;      }
16
;
17
; CHECK-NOT: The matrix multiplication pattern was detected
18
; CHECK-NOT: The tensor contraction pattern was detected
19

20
target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
21
target triple = "x86_64-unknown-linux-gnu"
22

23
define void @kernel_gemm(ptr %C, ptr %A, ptr %B, ptr %D) {
24
entry:
25
  br label %for.cond1.preheader
26

27
for.cond1.preheader:                              ; preds = %for.inc27, %entry
28
  %indvars.iv7 = phi i64 [ 0, %entry ], [ %indvars.iv.next8, %for.inc27 ]
29
  br label %for.cond4.preheader
30

31
for.cond4.preheader:                              ; preds = %for.inc24, %for.cond1.preheader
32
  %indvars.iv4 = phi i64 [ 0, %for.cond1.preheader ], [ %indvars.iv.next5, %for.inc24 ]
33
  %arrayidx22 = getelementptr inbounds [1024 x double], ptr %C, i64 %indvars.iv7, i64 %indvars.iv4
34
  br label %for.body6
35

36
for.body6:                                        ; preds = %for.body6, %for.cond4.preheader
37
  %indvars.iv = phi i64 [ 0, %for.cond4.preheader ], [ %indvars.iv.next, %for.body6 ]
38
  %arrayidx8 = getelementptr inbounds [1024 x double], ptr %A, i64 %indvars.iv7, i64 %indvars.iv
39
  %i = load double, ptr %arrayidx8, align 8
40
  %arrayidx12 = getelementptr inbounds [1024 x double], ptr %B, i64 %indvars.iv, i64 %indvars.iv4
41
  %i1 = load double, ptr %arrayidx12, align 8
42
  %mul = fmul double %i1, %i
43
  %arrayidx18 = getelementptr inbounds [1024 x [1024 x double]], ptr %D, i64 %indvars.iv7, i64 %indvars.iv4, i64 %indvars.iv
44
  %i2 = load double, ptr %arrayidx18, align 8
45
  %add = fadd double %i2, %mul
46
  store double %add, ptr %arrayidx18, align 8
47
  %i3 = load double, ptr %arrayidx22, align 8
48
  %add23 = fadd double %i3, %mul
49
  store double %add23, ptr %arrayidx22, align 8
50
  %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
51
  %exitcond.not = icmp eq i64 %indvars.iv.next, 1024
52
  br i1 %exitcond.not, label %for.inc24, label %for.body6
53

54
for.inc24:                                        ; preds = %for.body6
55
  %indvars.iv.next5 = add nuw nsw i64 %indvars.iv4, 1
56
  %exitcond6.not = icmp eq i64 %indvars.iv.next5, 1024
57
  br i1 %exitcond6.not, label %for.inc27, label %for.cond4.preheader
58

59
for.inc27:                                        ; preds = %for.inc24
60
  %indvars.iv.next8 = add nuw nsw i64 %indvars.iv7, 1
61
  %exitcond9.not = icmp eq i64 %indvars.iv.next8, 1024
62
  br i1 %exitcond9.not, label %for.end29, label %for.cond1.preheader
63

64
for.end29:                                        ; preds = %for.inc27
65
  ret void
66
}
67