test/Transforms/InstCombine/fabs.ll

   1 ; RUN: opt < %s -instcombine -S | FileCheck %s
   2
   3 ; Make sure all library calls are eliminated when the input is known positive.
   4
   5 declare float @fabsf(float)
   6 declare double @fabs(double)
   7 declare fp128 @fabsl(fp128)
   8
   9 define float @square_fabs_call_f32(float %x) {
  10   %mul = fmul float %x, %x
  11   %fabsf = tail call float @fabsf(float %mul)
  12   ret float %fabsf
  13
  14 ; CHECK-LABEL: square_fabs_call_f32(
  15 ; CHECK-NEXT: %mul = fmul float %x, %x
  16 ; CHECK-NEXT: ret float %mul
  17 }
  18
  19 define double @square_fabs_call_f64(double %x) {
  20   %mul = fmul double %x, %x
  21   %fabs = tail call double @fabs(double %mul)
  22   ret double %fabs
  23
  24 ; CHECK-LABEL: square_fabs_call_f64(
  25 ; CHECK-NEXT: %mul = fmul double %x, %x
  26 ; CHECK-NEXT: ret double %mul
  27 }
  28
  29 define fp128 @square_fabs_call_f128(fp128 %x) {
  30   %mul = fmul fp128 %x, %x
  31   %fabsl = tail call fp128 @fabsl(fp128 %mul)
  32   ret fp128 %fabsl
  33
  34 ; CHECK-LABEL: square_fabs_call_f128(
  35 ; CHECK-NEXT: %mul = fmul fp128 %x, %x
  36 ; CHECK-NEXT: ret fp128 %mul
  37 }
  38
  39 ; Make sure all intrinsic calls are eliminated when the input is known positive.
  40
  41 declare float @llvm.fabs.f32(float)
  42 declare double @llvm.fabs.f64(double)
  43 declare fp128 @llvm.fabs.f128(fp128)
  44
  45 define float @square_fabs_intrinsic_f32(float %x) {
  46   %mul = fmul float %x, %x
  47   %fabsf = tail call float @llvm.fabs.f32(float %mul)
  48   ret float %fabsf
  49
  50 ; CHECK-LABEL: square_fabs_intrinsic_f32(
  51 ; CHECK-NEXT: %mul = fmul float %x, %x
  52 ; CHECK-NEXT: ret float %mul
  53 }
  54
  55 define double @square_fabs_intrinsic_f64(double %x) {
  56   %mul = fmul double %x, %x
  57   %fabs = tail call double @llvm.fabs.f64(double %mul)
  58   ret double %fabs
  59
  60 ; CHECK-LABEL: square_fabs_intrinsic_f64(
  61 ; CHECK-NEXT: %mul = fmul double %x, %x
  62 ; CHECK-NEXT: ret double %mul
  63 }
  64
  65 define fp128 @square_fabs_intrinsic_f128(fp128 %x) {
  66   %mul = fmul fp128 %x, %x
  67   %fabsl = tail call fp128 @llvm.fabs.f128(fp128 %mul)
  68   ret fp128 %fabsl
  69
  70 ; CHECK-LABEL: square_fabs_intrinsic_f128(
  71 ; CHECK-NEXT: %mul = fmul fp128 %x, %x
  72 ; CHECK-NEXT: ret fp128 %mul
  73 }
  74
  75 ; Shrinking a library call to a smaller type should not be inhibited by nor inhibit the square optimization.
  76
  77 define float @square_fabs_shrink_call1(float %x) {
  78   %ext = fpext float %x to double
  79   %sq = fmul double %ext, %ext
  80   %fabs = call double @fabs(double %sq)
  81   %trunc = fptrunc double %fabs to float
  82   ret float %trunc
  83
  84 ; CHECK-LABEL: square_fabs_shrink_call1(
  85 ; CHECK-NEXT: %trunc = fmul float %x, %x
  86 ; CHECK-NEXT: ret float %trunc
  87 }
  88
  89 define float @square_fabs_shrink_call2(float %x) {
  90   %sq = fmul float %x, %x
  91   %ext = fpext float %sq to double
  92   %fabs = call double @fabs(double %ext)
  93   %trunc = fptrunc double %fabs to float
  94   ret float %trunc
  95
  96 ; CHECK-LABEL: square_fabs_shrink_call2(
  97 ; CHECK-NEXT: %sq = fmul float %x, %x
  98 ; CHECK-NEXT: ret float %sq
  99 }
 100