llvm/test/Transforms/LoopUnroll/X86/store_cost.ll

   1 ; REQUIRES: asserts
   2 ; RUN: opt -mcpu=core-avx2 -passes=loop-unroll -debug-only=loop-unroll -S -unroll-allow-partial < %s 2>&1 | FileCheck %s
   3
   4 target triple = "x86_64-unknown-linux-gnu"
   5
   6 ; CHECK: Loop Unroll: F[foo] Loop %loop.header
   7 ; CHECK:   Loop Size = 25
   8 ; CHECK: UNROLLING loop %loop.header by 2
   9 ; CHECK: Loop Unroll: F[foo] Loop %loop.2.header
  10 ; CHECK: Loop Size = 27
  11 ; CHECK-NOT: UNROLLING loop %loop.2.header
  12
  13 define void @foo(ptr %out) {
  14 entry:
  15   %0 = alloca [1024 x i32]
  16   %x0 = alloca [1024 x i32]
  17   %x01 = alloca [1024 x i32]
  18   %x02 = alloca [1024 x i32]
  19   %x03 = alloca [1024 x i32]
  20   %x04 = alloca [1024 x i32]
  21   %x05 = alloca [1024 x i32]
  22   %x06 = alloca [1024 x i32]
  23   br label %loop.header
  24
  25 loop.header:
  26   %counter = phi i32 [0, %entry], [%inc, %loop.inc]
  27   br label %loop.body
  28
  29 loop.body:
  30   %ptr = getelementptr [1024 x i32], ptr %0, i32 0, i32 %counter
  31   store i32 %counter, ptr %ptr
  32   %val = add i32 %counter, 5
  33   %xptr = getelementptr [1024 x i32], ptr %x0, i32 0, i32 %counter
  34   store i32 %val, ptr %xptr
  35   %val1 = add i32 %counter, 6
  36   %xptr1 = getelementptr [1024 x i32], ptr %x01, i32 0, i32 %counter
  37   store i32 %val1, ptr %xptr1
  38   %val2 = add i32 %counter, 7
  39   %xptr2 = getelementptr [1024 x i32], ptr %x02, i32 0, i32 %counter
  40   store i32 %val2, ptr %xptr2
  41   %val3 = add i32 %counter, 8
  42   %xptr3 = getelementptr [1024 x i32], ptr %x03, i32 0, i32 %counter
  43   store i32 %val3, ptr %xptr3
  44   %val4 = add i32 %counter, 9
  45   %xptr4 = getelementptr [1024 x i32], ptr %x04, i32 0, i32 %counter
  46   store i32 %val4, ptr %xptr4
  47   %val5 = add i32 %counter, 10
  48   %xptr5 = getelementptr [1024 x i32], ptr %x05, i32 0, i32 %counter
  49   store i32 %val5, ptr %xptr5
  50   br label %loop.inc
  51
  52 loop.inc:
  53   %inc = add i32 %counter, 2
  54   %1 = icmp sge i32 %inc, 1023
  55   br i1 %1, label  %exit.0, label %loop.header
  56
  57 exit.0:
  58   %2 = getelementptr [1024 x i32], ptr %0, i32 0, i32 5
  59   %3 = load i32, ptr %2
  60   store i32 %3, ptr %out
  61   br label %loop.2.header
  62
  63
  64 loop.2.header:
  65   %counter.2 = phi i32 [0, %exit.0], [%inc.2, %loop.2.inc]
  66   br label %loop.2.body
  67
  68 loop.2.body:
  69   %ptr.2 = getelementptr [1024 x i32], ptr %0, i32 0, i32 %counter.2
  70   store i32 %counter.2, ptr %ptr.2
  71   %val.2 = add i32 %counter.2, 5
  72   %xptr.2 = getelementptr [1024 x i32], ptr %x0, i32 0, i32 %counter.2
  73   store i32 %val.2, ptr %xptr.2
  74   %val1.2 = add i32 %counter.2, 6
  75   %xptr1.2 = getelementptr [1024 x i32], ptr %x01, i32 0, i32 %counter.2
  76   store i32 %val1, ptr %xptr1.2
  77   %val2.2 = add i32 %counter.2, 7
  78   %xptr2.2 = getelementptr [1024 x i32], ptr %x02, i32 0, i32 %counter.2
  79   store i32 %val2, ptr %xptr2.2
  80   %val3.2 = add i32 %counter.2, 8
  81   %xptr3.2 = getelementptr [1024 x i32], ptr %x03, i32 0, i32 %counter.2
  82   store i32 %val3.2, ptr %xptr3.2
  83   %val4.2 = add i32 %counter.2, 9
  84   %xptr4.2 = getelementptr [1024 x i32], ptr %x04, i32 0, i32 %counter.2
  85   store i32 %val4.2, ptr %xptr4.2
  86   %val5.2 = add i32 %counter.2, 10
  87   %xptr5.2 = getelementptr [1024 x i32], ptr %x05, i32 0, i32 %counter.2
  88   store i32 %val5.2, ptr %xptr5.2
  89   %xptr6.2 = getelementptr [1024 x i32], ptr %x06, i32 0, i32 %counter.2
  90   store i32 %val5.2, ptr %xptr6.2
  91   br label %loop.2.inc
  92
  93 loop.2.inc:
  94   %inc.2 = add i32 %counter.2, 2
  95   %4 = icmp sge i32 %inc.2, 1023
  96   br i1 %4, label  %exit.2, label %loop.2.header
  97
  98 exit.2:
  99   %x2 = getelementptr [1024 x i32], ptr %0, i32 0, i32 6
 100   %x3 = load i32, ptr %x2
 101   %out2 = getelementptr i32, ptr %out, i32 1
 102   store i32 %3, ptr %out2
 103   ret void
 104 }