llvm/test/Transforms/CodeExtractor/PartialInlineVarArg.ll

   1 ; RUN: opt < %s -passes=partial-inliner -S -skip-partial-inlining-cost-analysis | FileCheck %s
   2
   3 @stat = external global i32, align 4
   4
   5 define i32 @vararg(i32 %count, ...) {
   6 entry:
   7   %vargs = alloca ptr, align 8
   8   %stat1 = load i32, ptr @stat, align 4
   9   %cmp = icmp slt i32 %stat1, 0
  10   br i1 %cmp, label %bb2, label %bb1
  11
  12 bb1:                                              ; preds = %entry
  13   %vg1 = add nsw i32 %stat1, 1
  14   store i32 %vg1, ptr @stat, align 4
  15   call void @llvm.va_start(ptr %vargs)
  16   %va1 = va_arg ptr %vargs, i32
  17   call void @foo(i32 %count, i32 %va1) #2
  18   call void @llvm.va_end(ptr %vargs)
  19   br label %bb2
  20
  21 bb2:                                              ; preds = %bb1, %entry
  22   %res = phi i32 [ 1, %bb1 ], [ 0, %entry ]
  23   ret i32 %res
  24 }
  25
  26 declare void @foo(i32, i32)
  27 declare void @llvm.va_start(ptr)
  28 declare void @llvm.va_end(ptr)
  29
  30 define i32 @caller1(i32 %arg) {
  31 bb:
  32   %tmp = tail call i32 (i32, ...) @vararg(i32 %arg)
  33   ret i32 %tmp
  34 }
  35 ; CHECK-LABEL: @caller1
  36 ; CHECK: codeRepl.i:
  37 ; CHECK-NEXT:  call void (i32, ptr, i32, ...) @vararg.3.bb1(i32 %stat1.i, ptr %vargs.i, i32 %arg)
  38
  39 define i32 @caller2(i32 %arg, float %arg2) {
  40 bb:
  41   %tmp = tail call i32 (i32, ...) @vararg(i32 %arg, i32 10, float %arg2)
  42   ret i32 %tmp
  43 }
  44
  45 ; CHECK-LABEL: @caller2
  46 ; CHECK: codeRepl.i:
  47 ; CHECK-NEXT:  call void (i32, ptr, i32, ...) @vararg.3.bb1(i32 %stat1.i, ptr %vargs.i, i32 %arg, i32 10, float %arg2)
  48
  49 ; Test case to check that we do not extract a vararg function, if va_end is in
  50 ; a block that is not outlined.
  51 define i32 @vararg_not_legal(i32 %count, ...) {
  52 entry:
  53   %vargs = alloca ptr, align 8
  54   %stat1 = load i32, ptr @stat, align 4
  55   %cmp = icmp slt i32 %stat1, 0
  56   br i1 %cmp, label %bb2, label %bb1
  57
  58 bb1:                                              ; preds = %entry
  59   %vg1 = add nsw i32 %stat1, 1
  60   store i32 %vg1, ptr @stat, align 4
  61   call void @llvm.va_start(ptr %vargs)
  62   %va1 = va_arg ptr %vargs, i32
  63   call void @foo(i32 %count, i32 %va1)
  64   br label %bb2
  65
  66 bb2:                                              ; preds = %bb1, %entry
  67   %res = phi i32 [ 1, %bb1 ], [ 0, %entry ]
  68   %ptr = phi ptr [ %vargs, %bb1 ], [ %vargs, %entry]
  69   call void @llvm.va_end(ptr %ptr)
  70   ret i32 %res
  71 }
  72
  73 ; CHECK-LABEL: @caller3
  74 ; CHECK: tail call i32 (i32, ...) @vararg_not_legal(i32 %arg, i32 %arg)
  75 define i32 @caller3(i32 %arg) {
  76 bb:
  77   %res = tail call i32 (i32, ...) @vararg_not_legal(i32 %arg, i32 %arg)
  78   ret i32 %res
  79 }
  80
  81 declare ptr @err(ptr)
  82
  83 define signext i32 @vararg2(ptr %l, ...) {
  84 entry:
  85   %c = load i1, ptr %l
  86   br i1 %c, label %cleanup, label %cond.end
  87
  88 cond.end:                                         ; preds = %entry
  89   %call51 = call ptr @err(ptr nonnull %l)
  90   unreachable
  91
  92 cleanup:                                          ; preds = %entry
  93   ret i32 0
  94 }
  95
  96 define ptr @caller_with_signext(ptr %foo) {
  97 entry:
  98   %call1 = tail call signext i32 (ptr, ...) @vararg2(ptr %foo, i32 signext 8)
  99   unreachable
 100 }
 101
 102 ; CHECK-LABEL: @caller_with_signext
 103 ; CHECK: codeRepl.i:
 104 ; CHECK-NEXT:  call void (ptr, ...) @vararg2.1.cond.end(ptr %foo, i32 signext 8)