llvm/test/CodeGen/X86/vec_extract-mmx.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc < %s -mtriple=i686-unknown -mattr=+mmx,+sse2 | FileCheck %s --check-prefix=X32
   3 ; RUN: llc < %s -mtriple=x86_64-unknown -mattr=+mmx,+sse2 | FileCheck %s --check-prefix=X64
   4
   5 define i32 @test0(ptr %v4) nounwind {
   6 ; X32-LABEL: test0:
   7 ; X32:       # %bb.0: # %entry
   8 ; X32-NEXT:    movl {{[0-9]+}}(%esp), %eax
   9 ; X32-NEXT:    pshufw $238, (%eax), %mm0 # mm0 = mem[2,3,2,3]
  10 ; X32-NEXT:    movd %mm0, %eax
  11 ; X32-NEXT:    addl $32, %eax
  12 ; X32-NEXT:    retl
  13 ;
  14 ; X64-LABEL: test0:
  15 ; X64:       # %bb.0: # %entry
  16 ; X64-NEXT:    pshufw $238, (%rdi), %mm0 # mm0 = mem[2,3,2,3]
  17 ; X64-NEXT:    movd %mm0, %eax
  18 ; X64-NEXT:    addl $32, %eax
  19 ; X64-NEXT:    retq
  20 entry:
  21   %v5 = load <1 x i64>, ptr %v4, align 8
  22   %v12 = bitcast <1 x i64> %v5 to <4 x i16>
  23   %v13 = bitcast <4 x i16> %v12 to x86_mmx
  24   %v14 = tail call x86_mmx @llvm.x86.sse.pshuf.w(x86_mmx %v13, i8 -18)
  25   %v15 = bitcast x86_mmx %v14 to <4 x i16>
  26   %v16 = bitcast <4 x i16> %v15 to <1 x i64>
  27   %v17 = extractelement <1 x i64> %v16, i32 0
  28   %v18 = bitcast i64 %v17 to <2 x i32>
  29   %v19 = extractelement <2 x i32> %v18, i32 0
  30   %v20 = add i32 %v19, 32
  31   ret i32 %v20
  32 }
  33
  34 define i32 @test1(ptr nocapture readonly %ptr) nounwind {
  35 ; X32-LABEL: test1:
  36 ; X32:       # %bb.0: # %entry
  37 ; X32-NEXT:    movl {{[0-9]+}}(%esp), %eax
  38 ; X32-NEXT:    movd (%eax), %mm0
  39 ; X32-NEXT:    pshufw $232, %mm0, %mm0 # mm0 = mm0[0,2,2,3]
  40 ; X32-NEXT:    movd %mm0, %eax
  41 ; X32-NEXT:    emms
  42 ; X32-NEXT:    retl
  43 ;
  44 ; X64-LABEL: test1:
  45 ; X64:       # %bb.0: # %entry
  46 ; X64-NEXT:    movd (%rdi), %mm0
  47 ; X64-NEXT:    pshufw $232, %mm0, %mm0 # mm0 = mm0[0,2,2,3]
  48 ; X64-NEXT:    movd %mm0, %eax
  49 ; X64-NEXT:    emms
  50 ; X64-NEXT:    retq
  51 entry:
  52   %0 = load i32, ptr %ptr, align 4
  53   %1 = insertelement <2 x i32> undef, i32 %0, i32 0
  54   %2 = insertelement <2 x i32> %1, i32 0, i32 1
  55   %3 = bitcast <2 x i32> %2 to x86_mmx
  56   %4 = bitcast x86_mmx %3 to i64
  57   %5 = bitcast i64 %4 to <4 x i16>
  58   %6 = bitcast <4 x i16> %5 to x86_mmx
  59   %7 = tail call x86_mmx @llvm.x86.sse.pshuf.w(x86_mmx %6, i8 -24)
  60   %8 = bitcast x86_mmx %7 to <4 x i16>
  61   %9 = bitcast <4 x i16> %8 to <1 x i64>
  62   %10 = extractelement <1 x i64> %9, i32 0
  63   %11 = bitcast i64 %10 to <2 x i32>
  64   %12 = extractelement <2 x i32> %11, i32 0
  65   tail call void @llvm.x86.mmx.emms()
  66   ret i32 %12
  67 }
  68
  69 define i32 @test2(ptr nocapture readonly %ptr) nounwind {
  70 ; X32-LABEL: test2:
  71 ; X32:       # %bb.0: # %entry
  72 ; X32-NEXT:    movl {{[0-9]+}}(%esp), %eax
  73 ; X32-NEXT:    pshufw $232, (%eax), %mm0 # mm0 = mem[0,2,2,3]
  74 ; X32-NEXT:    movd %mm0, %eax
  75 ; X32-NEXT:    emms
  76 ; X32-NEXT:    retl
  77 ;
  78 ; X64-LABEL: test2:
  79 ; X64:       # %bb.0: # %entry
  80 ; X64-NEXT:    pshufw $232, (%rdi), %mm0 # mm0 = mem[0,2,2,3]
  81 ; X64-NEXT:    movd %mm0, %eax
  82 ; X64-NEXT:    emms
  83 ; X64-NEXT:    retq
  84 entry:
  85   %0 = load x86_mmx, ptr %ptr, align 8
  86   %1 = tail call x86_mmx @llvm.x86.sse.pshuf.w(x86_mmx %0, i8 -24)
  87   %2 = bitcast x86_mmx %1 to <4 x i16>
  88   %3 = bitcast <4 x i16> %2 to <1 x i64>
  89   %4 = extractelement <1 x i64> %3, i32 0
  90   %5 = bitcast i64 %4 to <2 x i32>
  91   %6 = extractelement <2 x i32> %5, i32 0
  92   tail call void @llvm.x86.mmx.emms()
  93   ret i32 %6
  94 }
  95
  96 define i32 @test3(x86_mmx %a) nounwind {
  97 ; X32-LABEL: test3:
  98 ; X32:       # %bb.0:
  99 ; X32-NEXT:    movd %mm0, %eax
 100 ; X32-NEXT:    retl
 101 ;
 102 ; X64-LABEL: test3:
 103 ; X64:       # %bb.0:
 104 ; X64-NEXT:    movd %mm0, %eax
 105 ; X64-NEXT:    retq
 106   %tmp0 = bitcast x86_mmx %a to <2 x i32>
 107   %tmp1 = extractelement <2 x i32> %tmp0, i32 0
 108   ret i32 %tmp1
 109 }
 110
 111 ; Verify we don't muck with extractelts from the upper lane.
 112 define i32 @test4(x86_mmx %a) nounwind {
 113 ; X32-LABEL: test4:
 114 ; X32:       # %bb.0:
 115 ; X32-NEXT:    movq2dq %mm0, %xmm0
 116 ; X32-NEXT:    pshufd {{.*#+}} xmm0 = xmm0[1,1,1,1]
 117 ; X32-NEXT:    movd %xmm0, %eax
 118 ; X32-NEXT:    retl
 119 ;
 120 ; X64-LABEL: test4:
 121 ; X64:       # %bb.0:
 122 ; X64-NEXT:    movq2dq %mm0, %xmm0
 123 ; X64-NEXT:    pshufd {{.*#+}} xmm0 = xmm0[1,1,1,1]
 124 ; X64-NEXT:    movd %xmm0, %eax
 125 ; X64-NEXT:    retq
 126   %tmp0 = bitcast x86_mmx %a to <2 x i32>
 127   %tmp1 = extractelement <2 x i32> %tmp0, i32 1
 128   ret i32 %tmp1
 129 }
 130
 131 declare x86_mmx @llvm.x86.sse.pshuf.w(x86_mmx, i8)
 132 declare void @llvm.x86.mmx.emms()