1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
; RUN: opt %loadPolly -polly-reschedule=0 -polly-opt-isl \
; RUN: -polly-pattern-matching-based-opts=true -polly-tc-opt=true \
; RUN: -debug -disable-output < %s 2>&1 | FileCheck %s
; REQUIRES: asserts
;
; for (i = 0; i < 1024; i++)
; for (j = 0; j < 1024; j++)
; for (l = 0; l < 64; ++l)
; for (w = 0; w < 64; ++w)
; C[i][j] += A[i][l][w] * B[w][j][l];
;
; CHECK: The tensor contraction pattern was detected
;
target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
target triple = "x86_64-unknown-linux-gnu"
define internal void @kernel_tc(i32 %ni, i32 %nj, i32 %nl, i32 %nq, i32 %nw, double %alpha, double %beta, [1024 x double]* %C, [64 x [64 x double]]* %A, [1024 x [64 x double]]* %B) {
entry:
br label %for.cond1.preheader
for.cond1.preheader: ; preds = %for.inc30, %entry
%indvars.iv43 = phi i64 [ 0, %entry ], [ %indvars.iv.next44, %for.inc30 ]
br label %for.cond4.preheader
for.cond4.preheader: ; preds = %for.inc27, %for.cond1.preheader
%indvars.iv40 = phi i64 [ 0, %for.cond1.preheader ], [ %indvars.iv.next41, %for.inc27 ]
br label %for.cond7.preheader
for.cond7.preheader: ; preds = %for.inc24, %for.cond4.preheader
%indvars.iv37 = phi i64 [ 0, %for.cond4.preheader ], [ %indvars.iv.next38, %for.inc24 ]
br label %for.body9
for.body9: ; preds = %for.body9, %for.cond7.preheader
%indvars.iv = phi i64 [ 0, %for.cond7.preheader ], [ %indvars.iv.next, %for.body9 ]
%arrayidx13 = getelementptr inbounds [64 x [64 x double]], [64 x [64 x double]]* %A, i64 %indvars.iv43, i64 %indvars.iv37, i64 %indvars.iv
%i = load double, double* %arrayidx13, align 8
%arrayidx19 = getelementptr inbounds [1024 x [64 x double]], [1024 x [64 x double]]* %B, i64 %indvars.iv, i64 %indvars.iv40, i64 %indvars.iv37
%i1 = load double, double* %arrayidx19, align 8
%mul = fmul fast double %i1, %i
%arrayidx23 = getelementptr inbounds [1024 x double], [1024 x double]* %C, i64 %indvars.iv43, i64 %indvars.iv40
%i2 = load double, double* %arrayidx23, align 8
%add = fadd fast double %i2, %mul
store double %add, double* %arrayidx23, align 8
%indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
%exitcond = icmp ne i64 %indvars.iv.next, 64
br i1 %exitcond, label %for.body9, label %for.inc24
for.inc24: ; preds = %for.body9
%indvars.iv.next38 = add nuw nsw i64 %indvars.iv37, 1
%exitcond39 = icmp ne i64 %indvars.iv.next38, 64
br i1 %exitcond39, label %for.cond7.preheader, label %for.inc27
for.inc27: ; preds = %for.inc24
%indvars.iv.next41 = add nuw nsw i64 %indvars.iv40, 1
%exitcond42 = icmp ne i64 %indvars.iv.next41, 1024
br i1 %exitcond42, label %for.cond4.preheader, label %for.inc30
for.inc30: ; preds = %for.inc27
%indvars.iv.next44 = add nuw nsw i64 %indvars.iv43, 1
%exitcond45 = icmp ne i64 %indvars.iv.next44, 1024
br i1 %exitcond45, label %for.cond1.preheader, label %for.end32
for.end32: ; preds = %for.inc30
ret void
}
|
