llvm/test/Transforms/CodeExtractor/LoopExtractor_alloca.ll

   1 ; RUN: opt -passes=debugify,loop-simplify,loop-extract -S < %s | FileCheck %s
   2
   3 ; This tests 2 cases:
   4 ; 1. loop1 should be extracted into a function, without extracting %v1 alloca.
   5 ; 2. loop2 should be extracted into a function, with the %v2 alloca.
   6 ;
   7 ; This used to produce an invalid IR, where `memcpy` will have a reference to
   8 ; the, now, external value (local to the extracted loop function).
   9
  10 ; CHECK-LABEL: define void @test()
  11 ; CHECK-NEXT: entry:
  12 ; CHECK-NEXT:   %v1 = alloca i32
  13 ; CHECK-NEXT:   call void @llvm.dbg.value(metadata ptr %v1
  14 ; CHECK-NEXT:   call void @llvm.memcpy.p0.p0.i64(ptr align 4 undef, ptr %v1, i64 4, i1 true)
  15
  16 ; CHECK-LABEL: define internal void @test.loop2()
  17 ; CHECK-NEXT: newFuncRoot:
  18 ; CHECK-NEXT:   %v2 = alloca i32
  19
  20 ; CHECK-LABEL: define internal void @test.loop1(ptr %v1)
  21 ; CHECK-NEXT: newFuncRoot:
  22 ; CHECK-NEXT:   br
  23
  24 define void @test() {
  25 entry:
  26   %v1 = alloca i32, align 4
  27   %v2 = alloca i32, align 4
  28   call void @llvm.memcpy.p0.p0.i64(ptr align 4 undef, ptr %v1, i64 4, i1 true)
  29   br label %loop1
  30
  31 loop1:
  32   call void @llvm.lifetime.start.p0(i64 4, ptr %v1)
  33   %r1 = call i32 @foo(ptr %v1)
  34   call void @llvm.lifetime.end.p0(i64 4, ptr %v1)
  35   %cmp1 = icmp ne i32 %r1, 0
  36   br i1 %cmp1, label %loop1, label %loop2
  37
  38 loop2:
  39   call void @llvm.lifetime.start.p0(i64 4, ptr %v2)
  40   %r2 = call i32 @foo(ptr %v2)
  41   call void @llvm.lifetime.end.p0(i64 4, ptr %v2)
  42   %cmp2 = icmp ne i32 %r2, 0
  43   br i1 %cmp2, label %loop2, label %exit
  44
  45 exit:
  46   ret void
  47 }
  48
  49 declare i32 @foo(ptr)
  50
  51 declare void @llvm.lifetime.start.p0(i64 immarg, ptr nocapture)
  52 declare void @llvm.lifetime.end.p0(i64 immarg, ptr nocapture)
  53 declare void @llvm.memcpy.p0.p0.i64(ptr noalias nocapture writeonly, ptr noalias nocapture readonly, i64, i1 immarg)