test/CodeGen/X86/pmul.ll

   1 ; RUN: llc < %s -mtriple=x86_64-unknown-unknown | FileCheck %s --check-prefix=ALL --check-prefix=SSE2
   2 ; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=sse4.1 | FileCheck %s --check-prefix=ALL --check-prefix=SSE41
   3
   4 define <4 x i32> @a(<4 x i32> %i) nounwind  {
   5 ; SSE2-LABEL: a:
   6 ; SSE2:       # BB#0: # %entry
   7 ; SSE2-NEXT:    movdqa {{.*#+}} xmm1 = [117,117,117,117]
   8 ; SSE2-NEXT:    pshufd {{.*#+}} xmm2 = xmm0[1,1,3,3]
   9 ; SSE2-NEXT:    pmuludq %xmm1, %xmm0
  10 ; SSE2-NEXT:    pmuludq %xmm1, %xmm2
  11 ; SSE2-NEXT:    shufps {{.*#+}} xmm0 = xmm0[0,2],xmm2[0,2]
  12 ; SSE2-NEXT:    shufps {{.*#+}} xmm0 = xmm0[0,2,1,3]
  13 ; SSE2-NEXT:    retq
  14 ;
  15 ; SSE41-LABEL: a:
  16 ; SSE41:       # BB#0: # %entry
  17 ; SSE41-NEXT:    pmulld {{.*}}(%rip), %xmm0
  18 ; SSE41-NEXT:    retq
  19 entry:
  20   %A = mul <4 x i32> %i, < i32 117, i32 117, i32 117, i32 117 >
  21   ret <4 x i32> %A
  22 }
  23
  24 define <2 x i64> @b(<2 x i64> %i) nounwind  {
  25 ; ALL-LABEL: b:
  26 ; ALL:       # BB#0: # %entry
  27 ; ALL-NEXT:    movdqa {{.*#+}} xmm1 = [117,117]
  28 ; ALL-NEXT:    movdqa %xmm0, %xmm2
  29 ; ALL-NEXT:    pmuludq %xmm1, %xmm2
  30 ; ALL-NEXT:    pxor %xmm3, %xmm3
  31 ; ALL-NEXT:    pmuludq %xmm0, %xmm3
  32 ; ALL-NEXT:    psllq $32, %xmm3
  33 ; ALL-NEXT:    paddq %xmm3, %xmm2
  34 ; ALL-NEXT:    psrlq $32, %xmm0
  35 ; ALL-NEXT:    pmuludq %xmm1, %xmm0
  36 ; ALL-NEXT:    psllq $32, %xmm0
  37 ; ALL-NEXT:    paddq %xmm2, %xmm0
  38 ; ALL-NEXT:    retq
  39 entry:
  40   %A = mul <2 x i64> %i, < i64 117, i64 117 >
  41   ret <2 x i64> %A
  42 }
  43
  44 define <4 x i32> @c(<4 x i32> %i, <4 x i32> %j) nounwind  {
  45 ; SSE2-LABEL: c:
  46 ; SSE2:       # BB#0: # %entry
  47 ; SSE2-NEXT:    pshufd {{.*#+}} xmm2 = xmm0[1,1,3,3]
  48 ; SSE2-NEXT:    pmuludq %xmm1, %xmm0
  49 ; SSE2-NEXT:    pshufd {{.*#+}} xmm1 = xmm1[1,1,3,3]
  50 ; SSE2-NEXT:    pmuludq %xmm2, %xmm1
  51 ; SSE2-NEXT:    shufps {{.*#+}} xmm0 = xmm0[0,2],xmm1[0,2]
  52 ; SSE2-NEXT:    shufps {{.*#+}} xmm0 = xmm0[0,2,1,3]
  53 ; SSE2-NEXT:    retq
  54 ;
  55 ; SSE41-LABEL: c:
  56 ; SSE41:       # BB#0: # %entry
  57 ; SSE41-NEXT:    pmulld %xmm1, %xmm0
  58 ; SSE41-NEXT:    retq
  59 entry:
  60   %A = mul <4 x i32> %i, %j
  61   ret <4 x i32> %A
  62 }
  63
  64 define <2 x i64> @d(<2 x i64> %i, <2 x i64> %j) nounwind  {
  65 ; ALL-LABEL: d:
  66 ; ALL:       # BB#0: # %entry
  67 ; ALL-NEXT:    movdqa %xmm0, %xmm2
  68 ; ALL-NEXT:    pmuludq %xmm1, %xmm2
  69 ; ALL-NEXT:    movdqa %xmm1, %xmm3
  70 ; ALL-NEXT:    psrlq $32, %xmm3
  71 ; ALL-NEXT:    pmuludq %xmm0, %xmm3
  72 ; ALL-NEXT:    psllq $32, %xmm3
  73 ; ALL-NEXT:    paddq %xmm3, %xmm2
  74 ; ALL-NEXT:    psrlq $32, %xmm0
  75 ; ALL-NEXT:    pmuludq %xmm1, %xmm0
  76 ; ALL-NEXT:    psllq $32, %xmm0
  77 ; ALL-NEXT:    paddq %xmm2, %xmm0
  78 ; ALL-NEXT:    retq
  79 entry:
  80   %A = mul <2 x i64> %i, %j
  81   ret <2 x i64> %A
  82 }
  83
  84 declare void @foo()
  85
  86 define <4 x i32> @e(<4 x i32> %i, <4 x i32> %j) nounwind  {
  87 ; SSE2-LABEL: e:
  88 ; SSE2:       # BB#0: # %entry
  89 ; SSE2-NEXT:    subq $40, %rsp
  90 ; SSE2-NEXT:    movaps %xmm1, {{[0-9]+}}(%rsp) # 16-byte Spill
  91 ; SSE2-NEXT:    movaps %xmm0, (%rsp) # 16-byte Spill
  92 ; SSE2-NEXT:    callq foo
  93 ; SSE2-NEXT:    movdqa (%rsp), %xmm0 # 16-byte Reload
  94 ; SSE2-NEXT:    pshufd {{.*#+}} xmm2 = xmm0[1,1,3,3]
  95 ; SSE2-NEXT:    movdqa {{[0-9]+}}(%rsp), %xmm1 # 16-byte Reload
  96 ; SSE2-NEXT:    pmuludq %xmm1, %xmm0
  97 ; SSE2-NEXT:    pshufd {{.*#+}} xmm1 = xmm1[1,1,3,3]
  98 ; SSE2-NEXT:    pmuludq %xmm2, %xmm1
  99 ; SSE2-NEXT:    shufps {{.*#+}} xmm0 = xmm0[0,2],xmm1[0,2]
 100 ; SSE2-NEXT:    shufps {{.*#+}} xmm0 = xmm0[0,2,1,3]
 101 ; SSE2-NEXT:    addq $40, %rsp
 102 ; SSE2-NEXT:    retq
 103 ;
 104 ; SSE41-LABEL: e:
 105 ; SSE41:       # BB#0: # %entry
 106 ; SSE41-NEXT:    subq $40, %rsp
 107 ; SSE41-NEXT:    movaps %xmm1, {{[0-9]+}}(%rsp) # 16-byte Spill
 108 ; SSE41-NEXT:    movaps %xmm0, (%rsp) # 16-byte Spill
 109 ; SSE41-NEXT:    callq foo
 110 ; SSE41-NEXT:    movdqa (%rsp), %xmm0 # 16-byte Reload
 111 ; SSE41-NEXT:    pmulld {{[0-9]+}}(%rsp), %xmm0 # 16-byte Folded Reload
 112 ; SSE41-NEXT:    addq $40, %rsp
 113 ; SSE41-NEXT:    retq
 114 entry:
 115   ; Use a call to force spills.
 116   call void @foo()
 117   %A = mul <4 x i32> %i, %j
 118   ret <4 x i32> %A
 119 }
 120
 121 define <2 x i64> @f(<2 x i64> %i, <2 x i64> %j) nounwind  {
 122 ; ALL-LABEL: f:
 123 ; ALL:       # BB#0: # %entry
 124 ; ALL-NEXT:    subq $40, %rsp
 125 ; ALL-NEXT:    movaps %xmm1, {{[0-9]+}}(%rsp) # 16-byte Spill
 126 ; ALL-NEXT:    movaps %xmm0, (%rsp) # 16-byte Spill
 127 ; ALL-NEXT:    callq foo
 128 ; ALL-NEXT:    movdqa (%rsp), %xmm0 # 16-byte Reload
 129 ; ALL-NEXT:    movdqa %xmm0, %xmm2
 130 ; ALL-NEXT:    movdqa {{[0-9]+}}(%rsp), %xmm3 # 16-byte Reload
 131 ; ALL-NEXT:    pmuludq %xmm3, %xmm2
 132 ; ALL-NEXT:    movdqa %xmm3, %xmm1
 133 ; ALL-NEXT:    psrlq $32, %xmm1
 134 ; ALL-NEXT:    pmuludq %xmm0, %xmm1
 135 ; ALL-NEXT:    psllq $32, %xmm1
 136 ; ALL-NEXT:    paddq %xmm1, %xmm2
 137 ; ALL-NEXT:    psrlq $32, %xmm0
 138 ; ALL-NEXT:    pmuludq %xmm3, %xmm0
 139 ; ALL-NEXT:    psllq $32, %xmm0
 140 ; ALL-NEXT:    paddq %xmm2, %xmm0
 141 ; ALL-NEXT:    addq $40, %rsp
 142 ; ALL-NEXT:    retq
 143 entry:
 144   ; Use a call to force spills.
 145   call void @foo()
 146   %A = mul <2 x i64> %i, %j
 147   ret <2 x i64> %A
 148 }