llvm/test/Transforms/FunctionSpecialization/function-specialization.ll

   1 ; RUN: opt -function-specialization -func-specialization-size-threshold=3 -S < %s | FileCheck %s
   2
   3 define i64 @main(i64 %x, i1 %flag) {
   4 ;
   5 ; CHECK-LABEL: @main(i64 %x, i1 %flag) {
   6 ; CHECK:      entry:
   7 ; CHECK-NEXT:   br i1 %flag, label %plus, label %minus
   8 ; CHECK:      plus:
   9 ; CHECK-NEXT:   [[TMP0:%.+]] = call i64 @compute.1(i64 %x, i64 (i64)* @plus)
  10 ; CHECK-NEXT:   br label %merge
  11 ; CHECK:      minus:
  12 ; CHECK-NEXT:   [[TMP1:%.+]] = call i64 @compute.2(i64 %x, i64 (i64)* @minus)
  13 ; CHECK-NEXT:   br label %merge
  14 ; CHECK:      merge:
  15 ; CHECK-NEXT:   [[TMP2:%.+]] = phi i64 [ [[TMP0]], %plus ], [ [[TMP1]], %minus ]
  16 ; CHECK-NEXT:   ret i64 [[TMP2]]
  17 ; CHECK-NEXT: }
  18 ;
  19 entry:
  20   br i1 %flag, label %plus, label %minus
  21
  22 plus:
  23   %tmp0 = call i64 @compute(i64 %x, i64 (i64)* @plus)
  24   br label %merge
  25
  26 minus:
  27   %tmp1 = call i64 @compute(i64 %x, i64 (i64)* @minus)
  28   br label %merge
  29
  30 merge:
  31   %tmp2 = phi i64 [ %tmp0, %plus ], [ %tmp1, %minus]
  32   ret i64 %tmp2
  33 }
  34
  35 ; CHECK-NOT: define internal i64 @compute(
  36 ;
  37 ; CHECK-LABEL: define internal i64 @compute.1(i64 %x, i64 (i64)* %binop) {
  38 ; CHECK-NEXT:  entry:
  39 ; CHECK-NEXT:    [[TMP0:%.+]] = call i64 @plus(i64 %x)
  40 ; CHECK-NEXT:    ret i64 [[TMP0]]
  41 ; CHECK-NEXT:  }
  42 ;
  43 ; CHECK-LABEL: define internal i64 @compute.2(i64 %x, i64 (i64)* %binop) {
  44 ; CHECK-NEXT:  entry:
  45 ; CHECK-NEXT:    [[TMP0:%.+]] = call i64 @minus(i64 %x)
  46 ; CHECK-NEXT:    ret i64 [[TMP0]]
  47 ; CHECK-NEXT:  }
  48 ;
  49 define internal i64 @compute(i64 %x, i64 (i64)* %binop) {
  50 entry:
  51   %tmp0 = call i64 %binop(i64 %x)
  52   ret i64 %tmp0
  53 }
  54
  55 define internal i64 @plus(i64 %x) {
  56 entry:
  57   %tmp0 = add i64 %x, 1
  58   ret i64 %tmp0
  59 }
  60
  61 define internal i64 @minus(i64 %x) {
  62 entry:
  63   %tmp0 = sub i64 %x, 1
  64   ret i64 %tmp0
  65 }