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

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