llvm/llvm/test/Transforms/CalledValuePropagation/simple-arguments.ll

; RUN: opt -passes=called-value-propagation -S < %s | FileCheck %s

target triple = "aarch64"


; This test checks that we propagate the functions through arguments and attach
; !callees metadata to the call. Such metadata can enable optimizations of this
; code sequence.
;
; For example, the code below a illustrates a contrived sort-like algorithm
; that accepts a pointer to a comparison function. Since the indirect call to
; the comparison function has only two targets, the call can be promoted to two
; direct calls using an if-then-else. The loop can then be unswitched and the
; called functions inlined. This essentially produces two loops, once
; specialized for each comparison.
;
; CHECK:  %tmp3 = call i1 %cmp(ptr %tmp1, ptr %tmp2), !callees ![[MD:[0-9]+]]
; CHECK: ![[MD]] = !{ptr @ugt, ptr @ule}
;
define void @test_argument(ptr %x, i64 %n, i1 %flag) {
entry:
  %tmp0 = sub i64 %n, 1
  br i1 %flag, label %then, label %else

then:
  call void @arrange_data(ptr %x, i64 %tmp0, ptr @ugt)
  br label %merge

else:
  call void @arrange_data(ptr %x, i64 %tmp0, ptr @ule)
  br label %merge

merge:
  ret void
}

define internal void @arrange_data(ptr %x, i64 %n, ptr %cmp) {
entry:
  %tmp0 = icmp eq i64 %n, 1
  br i1 %tmp0, label %merge, label %for.body

for.body:
  %i = phi i64 [ 0, %entry ], [ %i.next, %cmp.false ]
  %i.next = add nuw nsw i64 %i, 1
  %tmp1 = getelementptr inbounds i64, ptr %x, i64 %i
  %tmp2 = getelementptr inbounds i64, ptr %x, i64 %i.next
  %tmp3 = call i1 %cmp(ptr %tmp1, ptr %tmp2)
  br i1 %tmp3, label %cmp.true, label %cmp.false

cmp.true:
  call void @swap(ptr %tmp1, ptr %tmp2)
  br label %cmp.false

cmp.false:
  %cond = icmp slt i64 %i.next, %n
  br i1 %cond, label %for.body, label %for.end

for.end:
  %tmp4 = sub i64 %n, 1
  call void @arrange_data(ptr %x, i64 %tmp4, ptr %cmp)
  br label %merge

merge:
  ret void
}

define internal i1 @ugt(ptr %a, ptr %b) {
entry:
  %tmp0 = load i64, ptr %a
  %tmp1 = load i64, ptr %b
  %tmp2 = icmp ugt i64 %tmp0, %tmp1
  ret i1 %tmp2
}

define internal i1 @ule(ptr %a, ptr %b) {
entry:
  %tmp0 = load i64, ptr %a
  %tmp1 = load i64, ptr %b
  %tmp2 = icmp ule i64 %tmp0, %tmp1
  ret i1 %tmp2
}

declare void @swap(ptr, ptr)