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

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