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

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