test/CodeGen/X86/mulo-pow2.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=+avx | FileCheck %s --check-prefix=AVX
   3
   4 declare { <4 x i32>, <4 x i1> } @llvm.umul.with.overflow.v4i32(<4 x i32>, <4 x i32>)
   5 declare { <4 x i32>, <4 x i1> } @llvm.smul.with.overflow.v4i32(<4 x i32>, <4 x i32>)
   6
   7 define <4 x i32> @umul_v4i32_0(<4 x i32> %a, <4 x i32> %b) nounwind {
   8 ; AVX-LABEL: umul_v4i32_0:
   9 ; AVX:       # %bb.0:
  10 ; AVX-NEXT:    vxorps %xmm0, %xmm0, %xmm0
  11 ; AVX-NEXT:    retq
  12     %x = call { <4 x i32>, <4 x i1> } @llvm.umul.with.overflow.v4i32(<4 x i32> %a, <4 x i32> zeroinitializer)
  13     %y = extractvalue { <4 x i32>, <4 x i1> } %x, 0
  14     %z = extractvalue { <4 x i32>, <4 x i1> } %x, 1
  15     %u = select <4 x i1> %z, <4 x i32> %b, <4 x i32> %y
  16     ret <4 x i32> %u
  17 }
  18
  19 define <4 x i32> @umul_v4i32_1(<4 x i32> %a, <4 x i32> %b) nounwind {
  20 ; AVX-LABEL: umul_v4i32_1:
  21 ; AVX:       # %bb.0:
  22 ; AVX-NEXT:    retq
  23     %x = call { <4 x i32>, <4 x i1> } @llvm.umul.with.overflow.v4i32(<4 x i32> %a, <4 x i32> <i32 1, i32 1, i32 1, i32 1>)
  24     %y = extractvalue { <4 x i32>, <4 x i1> } %x, 0
  25     %z = extractvalue { <4 x i32>, <4 x i1> } %x, 1
  26     %u = select <4 x i1> %z, <4 x i32> %b, <4 x i32> %y
  27     ret <4 x i32> %u
  28 }
  29
  30 define <4 x i32> @umul_v4i32_2(<4 x i32> %a, <4 x i32> %b) nounwind {
  31 ; AVX-LABEL: umul_v4i32_2:
  32 ; AVX:       # %bb.0:
  33 ; AVX-NEXT:    vpaddd %xmm0, %xmm0, %xmm2
  34 ; AVX-NEXT:    vpmaxud %xmm0, %xmm2, %xmm0
  35 ; AVX-NEXT:    vpcmpeqd %xmm0, %xmm2, %xmm0
  36 ; AVX-NEXT:    vblendvps %xmm0, %xmm2, %xmm1, %xmm0
  37 ; AVX-NEXT:    retq
  38     %x = call { <4 x i32>, <4 x i1> } @llvm.umul.with.overflow.v4i32(<4 x i32> %a, <4 x i32> <i32 2, i32 2, i32 2, i32 2>)
  39     %y = extractvalue { <4 x i32>, <4 x i1> } %x, 0
  40     %z = extractvalue { <4 x i32>, <4 x i1> } %x, 1
  41     %u = select <4 x i1> %z, <4 x i32> %b, <4 x i32> %y
  42     ret <4 x i32> %u
  43 }
  44
  45 define <4 x i32> @umul_v4i32_8(<4 x i32> %a, <4 x i32> %b) nounwind {
  46 ; AVX-LABEL: umul_v4i32_8:
  47 ; AVX:       # %bb.0:
  48 ; AVX-NEXT:    vpslld $3, %xmm0, %xmm2
  49 ; AVX-NEXT:    vpsrld $3, %xmm2, %xmm3
  50 ; AVX-NEXT:    vpcmpeqd %xmm0, %xmm3, %xmm0
  51 ; AVX-NEXT:    vblendvps %xmm0, %xmm2, %xmm1, %xmm0
  52 ; AVX-NEXT:    retq
  53     %x = call { <4 x i32>, <4 x i1> } @llvm.umul.with.overflow.v4i32(<4 x i32> %a, <4 x i32> <i32 8, i32 8, i32 8, i32 8>)
  54     %y = extractvalue { <4 x i32>, <4 x i1> } %x, 0
  55     %z = extractvalue { <4 x i32>, <4 x i1> } %x, 1
  56     %u = select <4 x i1> %z, <4 x i32> %b, <4 x i32> %y
  57     ret <4 x i32> %u
  58 }
  59
  60 define <4 x i32> @umul_v4i32_2pow31(<4 x i32> %a, <4 x i32> %b) nounwind {
  61 ; AVX-LABEL: umul_v4i32_2pow31:
  62 ; AVX:       # %bb.0:
  63 ; AVX-NEXT:    vpslld $31, %xmm0, %xmm2
  64 ; AVX-NEXT:    vpsrld $31, %xmm2, %xmm3
  65 ; AVX-NEXT:    vpcmpeqd %xmm0, %xmm3, %xmm0
  66 ; AVX-NEXT:    vblendvps %xmm0, %xmm2, %xmm1, %xmm0
  67 ; AVX-NEXT:    retq
  68     %x = call { <4 x i32>, <4 x i1> } @llvm.umul.with.overflow.v4i32(<4 x i32> %a, <4 x i32> <i32 2147483648, i32 2147483648, i32 2147483648, i32 2147483648>)
  69     %y = extractvalue { <4 x i32>, <4 x i1> } %x, 0
  70     %z = extractvalue { <4 x i32>, <4 x i1> } %x, 1
  71     %u = select <4 x i1> %z, <4 x i32> %b, <4 x i32> %y
  72     ret <4 x i32> %u
  73 }
  74
  75 define <4 x i32> @smul_v4i32_0(<4 x i32> %a, <4 x i32> %b) nounwind {
  76 ; AVX-LABEL: smul_v4i32_0:
  77 ; AVX:       # %bb.0:
  78 ; AVX-NEXT:    vxorps %xmm0, %xmm0, %xmm0
  79 ; AVX-NEXT:    retq
  80     %x = call { <4 x i32>, <4 x i1> } @llvm.smul.with.overflow.v4i32(<4 x i32> %a, <4 x i32> zeroinitializer)
  81     %y = extractvalue { <4 x i32>, <4 x i1> } %x, 0
  82     %z = extractvalue { <4 x i32>, <4 x i1> } %x, 1
  83     %u = select <4 x i1> %z, <4 x i32> %b, <4 x i32> %y
  84     ret <4 x i32> %u
  85 }
  86
  87 define <4 x i32> @smul_v4i32_1(<4 x i32> %a, <4 x i32> %b) nounwind {
  88 ; AVX-LABEL: smul_v4i32_1:
  89 ; AVX:       # %bb.0:
  90 ; AVX-NEXT:    retq
  91     %x = call { <4 x i32>, <4 x i1> } @llvm.smul.with.overflow.v4i32(<4 x i32> %a, <4 x i32> <i32 1, i32 1, i32 1, i32 1>)
  92     %y = extractvalue { <4 x i32>, <4 x i1> } %x, 0
  93     %z = extractvalue { <4 x i32>, <4 x i1> } %x, 1
  94     %u = select <4 x i1> %z, <4 x i32> %b, <4 x i32> %y
  95     ret <4 x i32> %u
  96 }
  97
  98 define <4 x i32> @smul_v4i32_2(<4 x i32> %a, <4 x i32> %b) nounwind {
  99 ; AVX-LABEL: smul_v4i32_2:
 100 ; AVX:       # %bb.0:
 101 ; AVX-NEXT:    vpxor %xmm2, %xmm2, %xmm2
 102 ; AVX-NEXT:    vpcmpgtd %xmm0, %xmm2, %xmm3
 103 ; AVX-NEXT:    vpcmpeqd %xmm4, %xmm4, %xmm4
 104 ; AVX-NEXT:    vpxor %xmm4, %xmm3, %xmm3
 105 ; AVX-NEXT:    vpaddd %xmm0, %xmm0, %xmm0
 106 ; AVX-NEXT:    vpcmpgtd %xmm0, %xmm2, %xmm2
 107 ; AVX-NEXT:    vpxor %xmm4, %xmm2, %xmm2
 108 ; AVX-NEXT:    vpcmpeqd %xmm2, %xmm3, %xmm2
 109 ; AVX-NEXT:    vblendvps %xmm2, %xmm0, %xmm1, %xmm0
 110 ; AVX-NEXT:    retq
 111     %x = call { <4 x i32>, <4 x i1> } @llvm.smul.with.overflow.v4i32(<4 x i32> %a, <4 x i32> <i32 2, i32 2, i32 2, i32 2>)
 112     %y = extractvalue { <4 x i32>, <4 x i1> } %x, 0
 113     %z = extractvalue { <4 x i32>, <4 x i1> } %x, 1
 114     %u = select <4 x i1> %z, <4 x i32> %b, <4 x i32> %y
 115     ret <4 x i32> %u
 116 }
 117
 118 define <4 x i32> @smul_v4i32_8(<4 x i32> %a, <4 x i32> %b) nounwind {
 119 ; AVX-LABEL: smul_v4i32_8:
 120 ; AVX:       # %bb.0:
 121 ; AVX-NEXT:    vpslld $3, %xmm0, %xmm2
 122 ; AVX-NEXT:    vpsrad $3, %xmm2, %xmm3
 123 ; AVX-NEXT:    vpcmpeqd %xmm0, %xmm3, %xmm0
 124 ; AVX-NEXT:    vblendvps %xmm0, %xmm2, %xmm1, %xmm0
 125 ; AVX-NEXT:    retq
 126     %x = call { <4 x i32>, <4 x i1> } @llvm.smul.with.overflow.v4i32(<4 x i32> %a, <4 x i32> <i32 8, i32 8, i32 8, i32 8>)
 127     %y = extractvalue { <4 x i32>, <4 x i1> } %x, 0
 128     %z = extractvalue { <4 x i32>, <4 x i1> } %x, 1
 129     %u = select <4 x i1> %z, <4 x i32> %b, <4 x i32> %y
 130     ret <4 x i32> %u
 131 }
 132
 133 define <4 x i32> @smul_v4i32_2pow31(<4 x i32> %a, <4 x i32> %b) nounwind {
 134 ; AVX-LABEL: smul_v4i32_2pow31:
 135 ; AVX:       # %bb.0:
 136 ; AVX-NEXT:    vpslld $31, %xmm0, %xmm2
 137 ; AVX-NEXT:    vpsrld $31, %xmm2, %xmm3
 138 ; AVX-NEXT:    vpcmpeqd %xmm0, %xmm3, %xmm0
 139 ; AVX-NEXT:    vblendvps %xmm0, %xmm2, %xmm1, %xmm0
 140 ; AVX-NEXT:    retq
 141     %x = call { <4 x i32>, <4 x i1> } @llvm.smul.with.overflow.v4i32(<4 x i32> %a, <4 x i32> <i32 -2147483648, i32 -2147483648, i32 -2147483648, i32 -2147483648>)
 142     %y = extractvalue { <4 x i32>, <4 x i1> } %x, 0
 143     %z = extractvalue { <4 x i32>, <4 x i1> } %x, 1
 144     %u = select <4 x i1> %z, <4 x i32> %b, <4 x i32> %y
 145     ret <4 x i32> %u
 146 }