xref: /llvm-project/llvm/test/CodeGen/LoongArch/lasx/vselect.ll (revision 985d64b03accbed8500a85372d716367d89b61be)
1; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
2; RUN: llc --mtriple=loongarch64 --mattr=+lasx < %s | FileCheck %s
3
4define void @select_v32i8_imm(ptr %res, ptr %a0) nounwind {
5; CHECK-LABEL: select_v32i8_imm:
6; CHECK:       # %bb.0:
7; CHECK-NEXT:    xvld $xr0, $a1, 0
8; CHECK-NEXT:    xvrepli.h $xr1, -256
9; CHECK-NEXT:    xvbitseli.b $xr1, $xr0, 1
10; CHECK-NEXT:    xvst $xr1, $a0, 0
11; CHECK-NEXT:    ret
12  %v0 = load <32 x i8>, ptr %a0
13  %sel = select <32 x i1> <i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true>, <32 x i8> <i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1>, <32 x i8> %v0
14  store <32 x i8> %sel, ptr %res
15  ret void
16}
17
18define void @select_v32i8(ptr %res, ptr %a0, ptr %a1) nounwind {
19; CHECK-LABEL: select_v32i8:
20; CHECK:       # %bb.0:
21; CHECK-NEXT:    xvld $xr0, $a1, 0
22; CHECK-NEXT:    xvld $xr1, $a2, 0
23; CHECK-NEXT:    xvrepli.h $xr2, -256
24; CHECK-NEXT:    xvbitsel.v $xr0, $xr1, $xr0, $xr2
25; CHECK-NEXT:    xvst $xr0, $a0, 0
26; CHECK-NEXT:    ret
27  %v0 = load <32 x i8>, ptr %a0
28  %v1 = load <32 x i8>, ptr %a1
29  %sel = select <32 x i1> <i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true>, <32 x i8> %v0, <32 x i8> %v1
30  store <32 x i8> %sel, ptr %res
31  ret void
32}
33
34define void @select_v16i16(ptr %res, ptr %a0, ptr %a1) nounwind {
35; CHECK-LABEL: select_v16i16:
36; CHECK:       # %bb.0:
37; CHECK-NEXT:    xvld $xr0, $a1, 0
38; CHECK-NEXT:    xvld $xr1, $a2, 0
39; CHECK-NEXT:    lu12i.w $a1, -16
40; CHECK-NEXT:    xvreplgr2vr.w $xr2, $a1
41; CHECK-NEXT:    xvbitsel.v $xr0, $xr1, $xr0, $xr2
42; CHECK-NEXT:    xvst $xr0, $a0, 0
43; CHECK-NEXT:    ret
44  %v0 = load <16 x i16>, ptr %a0
45  %v1 = load <16 x i16>, ptr %a1
46  %sel = select <16 x i1> <i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true>, <16 x i16> %v0, <16 x i16> %v1
47  store <16 x i16> %sel, ptr %res
48  ret void
49}
50
51define void @select_v8i32(ptr %res, ptr %a0, ptr %a1) nounwind {
52; CHECK-LABEL: select_v8i32:
53; CHECK:       # %bb.0:
54; CHECK-NEXT:    xvld $xr0, $a1, 0
55; CHECK-NEXT:    xvld $xr1, $a2, 0
56; CHECK-NEXT:    ori $a1, $zero, 0
57; CHECK-NEXT:    lu32i.d $a1, -1
58; CHECK-NEXT:    xvreplgr2vr.d $xr2, $a1
59; CHECK-NEXT:    xvbitsel.v $xr0, $xr1, $xr0, $xr2
60; CHECK-NEXT:    xvst $xr0, $a0, 0
61; CHECK-NEXT:    ret
62  %v0 = load <8 x i32>, ptr %a0
63  %v1 = load <8 x i32>, ptr %a1
64  %sel = select <8 x i1> <i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true>, <8 x i32> %v0, <8 x i32> %v1
65  store <8 x i32> %sel, ptr %res
66  ret void
67}
68
69define void @select_v4i64(ptr %res, ptr %a0, ptr %a1) nounwind {
70; CHECK-LABEL: select_v4i64:
71; CHECK:       # %bb.0:
72; CHECK-NEXT:    xvld $xr0, $a1, 0
73; CHECK-NEXT:    xvld $xr1, $a2, 0
74; CHECK-NEXT:    pcalau12i $a1, %pc_hi20(.LCPI4_0)
75; CHECK-NEXT:    xvld $xr2, $a1, %pc_lo12(.LCPI4_0)
76; CHECK-NEXT:    xvbitsel.v $xr0, $xr1, $xr0, $xr2
77; CHECK-NEXT:    xvst $xr0, $a0, 0
78; CHECK-NEXT:    ret
79  %v0 = load <4 x i64>, ptr %a0
80  %v1 = load <4 x i64>, ptr %a1
81  %sel = select <4 x i1> <i1 true, i1 false, i1 true, i1 false>, <4 x i64> %v0, <4 x i64> %v1
82  store <4 x i64> %sel, ptr %res
83  ret void
84}
85