82 lines · plain
1; RUN: opt < %s -passes='print<loop-cache-cost>' -disable-output 2>&1 | FileCheck %s2 3target datalayout = "e-m:e-i64:64-n32:64"4target triple = "powerpc64le-unknown-linux-gnu"5 6; void matmul(long n, long m, long o, int A[n][m], int B[n][m], int C[n]) {7; for (long i = 0; i < n; i++) 8; for (long j = 0; j < m; j++) 9; for (long k = 0; k < o; k++) 10; C[i][j] = C[i][j] + A[i][k] * B[k][j];11; }12 13; CHECK:Loop 'for.i' has cost = 201000014; CHECK-NEXT:Loop 'for.k' has cost = 105000015; CHECK-NEXT:Loop 'for.j' has cost = 9000016 17define void @matmul(i64 %n, i64 %m, i64 %o, ptr %A, ptr %B, ptr %C) {18entry:19 br label %for.i20 21for.i: ; preds = %entry, %for.inc.i22 %i = phi i64 [ 0, %entry ], [ %i.next, %for.inc.i ]23 %muli = mul i64 %i, %m24 br label %for.j25 26for.j: ; preds = %for.i, %for.inc.j27 %j = phi i64 [ 0, %for.i ], [ %j.next, %for.inc.j ]28 %addj = add i64 %muli, %j29 %mulj = mul i64 %addj, %o30 br label %for.k31 32for.k: ; preds = %for.j, %for.inc.k33 %k = phi i64 [ 0, %for.j ], [ %k.next, %for.inc.k ]34 35 ; A[i][k]36 %arrayidx3 = add i64 %k, %muli37 %arrayidx4 = getelementptr inbounds i32, ptr %A, i64 %arrayidx338 %elem_A = load i32, ptr %arrayidx4, align 439 40 ; B[k][j]41 %mulk = mul i64 %k, %o42 %arrayidx5 = add i64 %j, %mulk43 %arrayidx6 = getelementptr inbounds i32, ptr %B, i64 %arrayidx544 %elem_B = load i32, ptr %arrayidx6, align 445 46 ; C[i][k]47 %arrayidx7 = add i64 %j, %muli48 %arrayidx8 = getelementptr inbounds i32, ptr %C, i64 %arrayidx749 %elem_C = load i32, ptr %arrayidx8, align 450 51 ; C[i][j] = C[i][j] + A[i][k] * B[k][j];52 %mul = mul nsw i32 %elem_A, %elem_B53 %add = add nsw i32 %elem_C, %mul54 store i32 %add, ptr %arrayidx8, align 455 56 br label %for.inc.k57 58for.inc.k: ; preds = %for.k59 %k.next = add nuw nsw i64 %k, 160 %exitcond = icmp ne i64 %k.next, %o61 br i1 %exitcond, label %for.k, label %for.end62 63for.end: ; preds = %for.inc64 br label %for.inc.j65 66for.inc.j: ; preds = %for.end67 %j.next = add nuw nsw i64 %j, 168 %exitcond5 = icmp ne i64 %j.next, %m69 br i1 %exitcond5, label %for.j, label %for.end2370 71for.end23: ; preds = %for.inc.j72 br label %for.inc.i73 74for.inc.i: ; preds = %for.end2375 %i.next = add nuw nsw i64 %i, 176 %exitcond8 = icmp ne i64 %i.next, %n77 br i1 %exitcond8, label %for.i, label %for.end2678 79for.end26: ; preds = %for.inc.i80 ret void81}82