llvm/test/Transforms/InstSimplify/and-or-icmp-ctpop.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt < %s -passes=instsimplify -S | FileCheck %s
   3
   4 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
   5 ;
   6 ; (ctpop(X) == N) || (X != 0) --> X != 0
   7 ; where N > 0
   8 ;
   9 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
  10
  11 declare i32 @llvm.ctpop.i32(i32)
  12
  13 define i1 @eq_or_non_0(i32 %x) {
  14 ; CHECK-LABEL: @eq_or_non_0(
  15 ; CHECK-NEXT:    [[NOTZERO:%.*]] = icmp ne i32 [[X:%.*]], 0
  16 ; CHECK-NEXT:    ret i1 [[NOTZERO]]
  17 ;
  18   %t0 = tail call i32 @llvm.ctpop.i32(i32 %x)
  19   %cmp = icmp eq i32 %t0, 10
  20   %notzero = icmp ne i32 %x, 0
  21   %r = or i1 %notzero, %cmp
  22   ret i1 %r
  23 }
  24
  25 declare <2 x i32> @llvm.ctpop.v2i32(<2 x i32>)
  26
  27 define <2 x i1> @eq_or_non_0_commute(<2 x i32> %x) {
  28 ; CHECK-LABEL: @eq_or_non_0_commute(
  29 ; CHECK-NEXT:    [[NOTZERO:%.*]] = icmp ne <2 x i32> [[X:%.*]], zeroinitializer
  30 ; CHECK-NEXT:    ret <2 x i1> [[NOTZERO]]
  31 ;
  32   %t0 = tail call <2 x i32> @llvm.ctpop.v2i32(<2 x i32> %x)
  33   %cmp = icmp eq <2 x i32> %t0, <i32 20, i32 20>
  34   %notzero = icmp ne <2 x i32> %x, <i32 0, i32 0>
  35   %r = or <2 x i1> %cmp, %notzero
  36   ret <2 x i1> %r
  37 }
  38
  39 ; Negative test - wrong predicate
  40
  41 define i1 @eq_or_non_0_wrong_pred1(i32 %x) {
  42 ; CHECK-LABEL: @eq_or_non_0_wrong_pred1(
  43 ; CHECK-NEXT:    [[T0:%.*]] = tail call i32 @llvm.ctpop.i32(i32 [[X:%.*]])
  44 ; CHECK-NEXT:    [[CMP:%.*]] = icmp ne i32 [[T0]], 10
  45 ; CHECK-NEXT:    [[NOTZERO:%.*]] = icmp ne i32 [[X]], 0
  46 ; CHECK-NEXT:    [[R:%.*]] = or i1 [[NOTZERO]], [[CMP]]
  47 ; CHECK-NEXT:    ret i1 [[R]]
  48 ;
  49   %t0 = tail call i32 @llvm.ctpop.i32(i32 %x)
  50   %cmp = icmp ne i32 %t0, 10
  51   %notzero = icmp ne i32 %x, 0
  52   %r = or i1 %notzero, %cmp
  53   ret i1 %r
  54 }
  55
  56 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
  57 ;
  58 ; (ctpop(X) != N) && (X == 0) --> X == 0
  59 ; where N > 0
  60 ;
  61 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
  62
  63 define i1 @ne_and_is_0(i32 %x) {
  64 ; CHECK-LABEL: @ne_and_is_0(
  65 ; CHECK-NEXT:    [[ISZERO:%.*]] = icmp eq i32 [[X:%.*]], 0
  66 ; CHECK-NEXT:    ret i1 [[ISZERO]]
  67 ;
  68   %t0 = tail call i32 @llvm.ctpop.i32(i32 %x)
  69   %cmp = icmp ne i32 %t0, 10
  70   %iszero = icmp eq i32 %x, 0
  71   %r = and i1 %iszero, %cmp
  72   ret i1 %r
  73 }
  74
  75 define <2 x i1> @ne_and_is_0_commute(<2 x i32> %x) {
  76 ; CHECK-LABEL: @ne_and_is_0_commute(
  77 ; CHECK-NEXT:    [[ISZERO:%.*]] = icmp eq <2 x i32> [[X:%.*]], zeroinitializer
  78 ; CHECK-NEXT:    ret <2 x i1> [[ISZERO]]
  79 ;
  80   %t0 = tail call <2 x i32> @llvm.ctpop.v2i32(<2 x i32> %x)
  81   %cmp = icmp ne <2 x i32> %t0, <i32 20, i32 20>
  82   %iszero = icmp eq <2 x i32> %x, <i32 0, i32 0>
  83   %r = and <2 x i1> %cmp, %iszero
  84   ret <2 x i1> %r
  85 }
  86
  87 ; Negative test - wrong predicate
  88
  89 define i1 @ne_and_is_0_wrong_pred1(i32 %x) {
  90 ; CHECK-LABEL: @ne_and_is_0_wrong_pred1(
  91 ; CHECK-NEXT:    [[T0:%.*]] = tail call i32 @llvm.ctpop.i32(i32 [[X:%.*]])
  92 ; CHECK-NEXT:    [[CMP:%.*]] = icmp ne i32 [[T0]], 10
  93 ; CHECK-NEXT:    [[ISZERO:%.*]] = icmp eq i32 [[X]], 0
  94 ; CHECK-NEXT:    [[R:%.*]] = or i1 [[ISZERO]], [[CMP]]
  95 ; CHECK-NEXT:    ret i1 [[R]]
  96 ;
  97   %t0 = tail call i32 @llvm.ctpop.i32(i32 %x)
  98   %cmp = icmp ne i32 %t0, 10
  99   %iszero = icmp eq i32 %x, 0
 100   %r = or i1 %iszero, %cmp
 101   ret i1 %r
 102 }