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

   1 ; RUN: opt -function-specialization -force-function-specialization \
   2 ; RUN:   -func-specialization-max-constants=2 -S < %s | FileCheck %s
   3
   4 ; RUN: opt -function-specialization -force-function-specialization \
   5 ; RUN:   -func-specialization-max-constants=1 -S < %s | FileCheck %s --check-prefix=CONST1
   6
   7 target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128"
   8
   9 @A = external dso_local constant i32, align 4
  10 @B = external dso_local constant i32, align 4
  11 @C = external dso_local constant i32, align 4
  12 @D = external dso_local constant i32, align 4
  13
  14 define dso_local i32 @bar(i32 %x, i32 %y) {
  15 entry:
  16   %tobool = icmp ne i32 %x, 0
  17   br i1 %tobool, label %if.then, label %if.else
  18
  19 if.then:
  20   %call = call i32 @foo(i32 %x, i32* @A, i32* @C)
  21   br label %return
  22
  23 if.else:
  24   %call1 = call i32 @foo(i32 %y, i32* @B, i32* @D)
  25   br label %return
  26
  27 return:
  28   %retval.0 = phi i32 [ %call, %if.then ], [ %call1, %if.else ]
  29   ret i32 %retval.0
  30 }
  31
  32 define internal i32 @foo(i32 %x, i32* %b, i32* %c) {
  33 entry:
  34   %0 = load i32, i32* %b, align 4
  35   %add = add nsw i32 %x, %0
  36   %1 = load i32, i32* %c, align 4
  37   %add1 = add nsw i32 %add, %1
  38   ret i32 %add1
  39 }
  40
  41 ; CONST1-NOT: define internal i32 @foo.1(i32 %x, i32* %b, i32* %c)
  42 ; CONST1-NOT: define internal i32 @foo.2(i32 %x, i32* %b, i32* %c)
  43
  44 ; CHECK:        define internal i32 @foo.1(i32 %x, i32* %b, i32* %c) {
  45 ; CHECK-NEXT:   entry:
  46 ; CHECK-NEXT:     %0 = load i32, i32* @A, align 4
  47 ; CHECK-NEXT:     %add = add nsw i32 %x, %0
  48 ; CHECK-NEXT:     %1 = load i32, i32* %c, align 4
  49 ; CHECK-NEXT:     %add1 = add nsw i32 %add, %1
  50 ; CHECK-NEXT:     ret i32 %add1
  51 ; CHECK-NEXT:   }
  52
  53 ; CHECK: define internal i32 @foo.2(i32 %x, i32* %b, i32* %c) {
  54 ; CHECK-NEXT:   entry:
  55 ; CHECK-NEXT:     %0 = load i32, i32* @B, align 4
  56 ; CHECK-NEXT:     %add = add nsw i32 %x, %0
  57 ; CHECK-NEXT:     %1 = load i32, i32* %c, align 4
  58 ; CHECK-NEXT:     %add1 = add nsw i32 %add, %1
  59 ; CHECK-NEXT:     ret i32 %add1
  60 ; CHECK-NEXT:   }