llvm/test/Transforms/FunctionSpecialization/literal-const.ll

   1 ; RUN: opt -S --passes="ipsccp<func-spec>" \
   2 ; RUN:        -force-specialization < %s | FileCheck %s -check-prefix CHECK-NOLIT
   3 ; RUN: opt -S --passes="ipsccp<func-spec>" \
   4 ; RUN:        -funcspec-for-literal-constant \
   5 ; RUN:        -force-specialization < %s | FileCheck %s -check-prefix CHECK-LIT
   6
   7 define i32 @f0(i32 noundef %x) {
   8 entry:
   9   %call = tail call i32 @neg(i32 noundef %x, i1 noundef zeroext false)
  10   ret i32 %call
  11 }
  12
  13 define i32 @f1(i32 noundef %x) {
  14 entry:
  15   %call = tail call i32 @neg(i32 noundef %x, i1 noundef zeroext true)
  16   ret i32 %call
  17 }
  18
  19 define i32 @g0(i32 noundef %x) {
  20 entry:
  21   %call = tail call i32 @add(i32 noundef %x, i32 noundef 1)
  22   ret i32 %call
  23 }
  24
  25 define i32 @g1(i32 noundef %x) {
  26 entry:
  27   %call = tail call i32 @add(i32 noundef %x, i32 noundef 2)
  28   ret i32 %call
  29 }
  30
  31 define float @h0(float noundef %x) {
  32 entry:
  33   %call = tail call float @addf(float noundef %x, float noundef 1.000000e+00)
  34   ret float %call
  35 }
  36
  37 define float @h1(float noundef %x) {
  38 entry:
  39   %call = tail call float @addf(float noundef %x, float noundef 2.000000e+00)
  40   ret float %call
  41 }
  42
  43 define internal i32 @neg(i32 noundef %x, i1 noundef zeroext %b) {
  44 entry:
  45   %sub = sub nsw i32 0, %x
  46   %cond = select i1 %b, i32 %sub, i32 %x
  47   ret i32 %cond
  48 }
  49
  50 define internal i32 @add(i32 noundef %x, i32 noundef %y) {
  51 entry:
  52   %add = add nsw i32 %y, %x
  53   ret i32 %add
  54 }
  55
  56 define internal float @addf(float noundef %x, float noundef %y) {
  57 entry:
  58   %add = fadd float %x, %y
  59   ret float %add
  60 }
  61
  62
  63 ; Check no functions were specialised.
  64 ; CHECK-NOLIT-NOT: @neg.
  65 ; CHECK-NOLIT-NOT: @add.
  66 ; CHECK-NOLIT-NOT: @addf.
  67
  68 ; CHECK-LIT-LABEL: define i32 @f0
  69 ; CHECK-LIT: call i32 @neg.specialized.[[#A:]]
  70
  71 ; CHECK-LIT-LABEL: define i32 @f1
  72 ; CHECK-LIT: call i32 @neg.specialized.[[#B:]]
  73
  74 ; CHECK-LIT-LABEL: define i32 @g0
  75 ; CHECK-LIT: call i32 @add.specialized.[[#C:]]
  76
  77 ; CHECK-LIT-LABEL: define i32 @g1
  78 ; CHECK-LIT: call i32 @add.specialized.[[#D:]]
  79
  80 ; CHECK-LIT-LABEL: define float @h0
  81 ; CHECK-LIT: call float @addf.specialized.[[#E:]]
  82
  83 ; CHECK-LIT-LABEL: define float @h1
  84 ; CHECK-LIT: call float @addf.specialized.[[#F:]]
  85
  86 ; Check all of `neg`, `add`, and `addf` were specialised.
  87 ; CHECK-LIT-DAG: @neg.specialized.[[#A]]
  88 ; CHECK-LIT-DAG: @neg.specialized.[[#B]]
  89 ; CHECK-LIT-DAG: @add.specialized.[[#C]]
  90 ; CHECK-LIT-DAG: @add.specialized.[[#D]]
  91 ; CHECK-LIT-DAG: @addf.specialized.[[#E]]
  92 ; CHECK-LIT-DAG: @addf.specialized.[[#F]]