xref: /llvm-project/llvm/test/CodeGen/X86/vec_extract.ll (revision eb523a4d272e81c8f7bf48da3923ed502f41c187)
1; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
2; RUN: llc < %s -mtriple=i686-unknown-linux-gnu -mattr=+sse2,-sse4.1 | FileCheck %s --check-prefix=X86
3; RUN: llc < %s -mtriple=x86_64-unknown-linux-gnu -mattr=+sse2,-sse4.1 | FileCheck %s --check-prefix=X64
4
5define void @test1(ptr %F, ptr %f) nounwind {
6; X86-LABEL: test1:
7; X86:       # %bb.0: # %entry
8; X86-NEXT:    movl {{[0-9]+}}(%esp), %eax
9; X86-NEXT:    movl {{[0-9]+}}(%esp), %ecx
10; X86-NEXT:    movss {{.*#+}} xmm0 = mem[0],zero,zero,zero
11; X86-NEXT:    addss %xmm0, %xmm0
12; X86-NEXT:    movss %xmm0, (%eax)
13; X86-NEXT:    retl
14;
15; X64-LABEL: test1:
16; X64:       # %bb.0: # %entry
17; X64-NEXT:    movss {{.*#+}} xmm0 = mem[0],zero,zero,zero
18; X64-NEXT:    addss %xmm0, %xmm0
19; X64-NEXT:    movss %xmm0, (%rsi)
20; X64-NEXT:    retq
21entry:
22  %tmp = load <4 x float>, ptr %F
23  %tmp7 = fadd <4 x float> %tmp, %tmp
24  %tmp2 = extractelement <4 x float> %tmp7, i32 0
25  store float %tmp2, ptr %f
26  ret void
27}
28
29define float @test2(ptr %F, ptr %f) nounwind {
30; X86-LABEL: test2:
31; X86:       # %bb.0: # %entry
32; X86-NEXT:    pushl %eax
33; X86-NEXT:    movl {{[0-9]+}}(%esp), %eax
34; X86-NEXT:    movaps (%eax), %xmm0
35; X86-NEXT:    addps %xmm0, %xmm0
36; X86-NEXT:    movhlps {{.*#+}} xmm0 = xmm0[1,1]
37; X86-NEXT:    movss %xmm0, (%esp)
38; X86-NEXT:    flds (%esp)
39; X86-NEXT:    popl %eax
40; X86-NEXT:    retl
41;
42; X64-LABEL: test2:
43; X64:       # %bb.0: # %entry
44; X64-NEXT:    movaps (%rdi), %xmm0
45; X64-NEXT:    addps %xmm0, %xmm0
46; X64-NEXT:    movhlps {{.*#+}} xmm0 = xmm0[1,1]
47; X64-NEXT:    retq
48entry:
49  %tmp = load <4 x float>, ptr %F
50  %tmp7 = fadd <4 x float> %tmp, %tmp
51  %tmp2 = extractelement <4 x float> %tmp7, i32 2
52  ret float %tmp2
53}
54
55define void @test3(ptr %R, ptr %P1) nounwind {
56; X86-LABEL: test3:
57; X86:       # %bb.0: # %entry
58; X86-NEXT:    movl {{[0-9]+}}(%esp), %eax
59; X86-NEXT:    movl {{[0-9]+}}(%esp), %ecx
60; X86-NEXT:    movaps (%ecx), %xmm0
61; X86-NEXT:    shufps {{.*#+}} xmm0 = xmm0[3,3,3,3]
62; X86-NEXT:    movss %xmm0, (%eax)
63; X86-NEXT:    retl
64;
65; X64-LABEL: test3:
66; X64:       # %bb.0: # %entry
67; X64-NEXT:    movaps (%rsi), %xmm0
68; X64-NEXT:    shufps {{.*#+}} xmm0 = xmm0[3,3,3,3]
69; X64-NEXT:    movss %xmm0, (%rdi)
70; X64-NEXT:    retq
71entry:
72  %X = load <4 x float>, ptr %P1
73  %tmp = extractelement <4 x float> %X, i32 3
74  store float %tmp, ptr %R
75  ret void
76}
77
78define double @test4(double %A) nounwind {
79; X86-LABEL: test4:
80; X86:       # %bb.0: # %entry
81; X86-NEXT:    subl $12, %esp
82; X86-NEXT:    calll foo@PLT
83; X86-NEXT:    unpckhpd {{.*#+}} xmm0 = xmm0[1,1]
84; X86-NEXT:    addsd {{[0-9]+}}(%esp), %xmm0
85; X86-NEXT:    movsd %xmm0, (%esp)
86; X86-NEXT:    fldl (%esp)
87; X86-NEXT:    addl $12, %esp
88; X86-NEXT:    retl
89;
90; X64-LABEL: test4:
91; X64:       # %bb.0: # %entry
92; X64-NEXT:    pushq %rax
93; X64-NEXT:    movsd %xmm0, (%rsp) # 8-byte Spill
94; X64-NEXT:    callq foo@PLT
95; X64-NEXT:    unpckhpd {{.*#+}} xmm0 = xmm0[1,1]
96; X64-NEXT:    addsd (%rsp), %xmm0 # 8-byte Folded Reload
97; X64-NEXT:    popq %rax
98; X64-NEXT:    retq
99entry:
100  %tmp1 = call <2 x double> @foo( )
101  %tmp2 = extractelement <2 x double> %tmp1, i32 1
102  %tmp3 = fadd double %tmp2, %A
103  ret double %tmp3
104}
105declare <2 x double> @foo()
106
107; OSS-Fuzz #15662
108; https://bugs.chromium.org/p/oss-fuzz/issues/detail?id=15662
109define <4 x i32> @ossfuzz15662(ptr %in) {
110; X86-LABEL: ossfuzz15662:
111; X86:       # %bb.0:
112; X86-NEXT:    xorps %xmm0, %xmm0
113; X86-NEXT:    movaps %xmm0, (%eax)
114; X86-NEXT:    retl
115;
116; X64-LABEL: ossfuzz15662:
117; X64:       # %bb.0:
118; X64-NEXT:    xorps %xmm0, %xmm0
119; X64-NEXT:    movaps %xmm0, (%rax)
120; X64-NEXT:    retq
121   %C10 = icmp ule i1 false, false
122   %C3 = icmp ule i1 true, undef
123   %B = sdiv i1 %C10, %C3
124   %I = insertelement <4 x i32> zeroinitializer, i32 0, i1 %B
125   store <4 x i32> %I, ptr undef
126   ret <4 x i32> zeroinitializer
127}
128