xref: /llvm-project/llvm/test/CodeGen/LoongArch/ir-instruction/lshr.ll (revision 9d4f7f44b64d87d1068859906f43b7ce03a7388b)
1; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
2; RUN: llc --mtriple=loongarch32 -mattr=+d < %s | FileCheck %s --check-prefix=LA32
3; RUN: llc --mtriple=loongarch64 -mattr=+d < %s | FileCheck %s --check-prefix=LA64
4
5;; Exercise the 'lshr' LLVM IR: https://llvm.org/docs/LangRef.html#lshr-instruction
6
7define i1 @lshr_i1(i1 %x, i1 %y) {
8; LA32-LABEL: lshr_i1:
9; LA32:       # %bb.0:
10; LA32-NEXT:    ret
11;
12; LA64-LABEL: lshr_i1:
13; LA64:       # %bb.0:
14; LA64-NEXT:    ret
15  %lshr = lshr i1 %x, %y
16  ret i1 %lshr
17}
18
19define i8 @lshr_i8(i8 %x, i8 %y) {
20; LA32-LABEL: lshr_i8:
21; LA32:       # %bb.0:
22; LA32-NEXT:    andi $a0, $a0, 255
23; LA32-NEXT:    srl.w $a0, $a0, $a1
24; LA32-NEXT:    ret
25;
26; LA64-LABEL: lshr_i8:
27; LA64:       # %bb.0:
28; LA64-NEXT:    andi $a0, $a0, 255
29; LA64-NEXT:    srl.d $a0, $a0, $a1
30; LA64-NEXT:    ret
31  %lshr = lshr i8 %x, %y
32  ret i8 %lshr
33}
34
35define i16 @lshr_i16(i16 %x, i16 %y) {
36; LA32-LABEL: lshr_i16:
37; LA32:       # %bb.0:
38; LA32-NEXT:    bstrpick.w $a0, $a0, 15, 0
39; LA32-NEXT:    srl.w $a0, $a0, $a1
40; LA32-NEXT:    ret
41;
42; LA64-LABEL: lshr_i16:
43; LA64:       # %bb.0:
44; LA64-NEXT:    bstrpick.d $a0, $a0, 15, 0
45; LA64-NEXT:    srl.d $a0, $a0, $a1
46; LA64-NEXT:    ret
47  %lshr = lshr i16 %x, %y
48  ret i16 %lshr
49}
50
51define i32 @lshr_i32(i32 %x, i32 %y) {
52; LA32-LABEL: lshr_i32:
53; LA32:       # %bb.0:
54; LA32-NEXT:    srl.w $a0, $a0, $a1
55; LA32-NEXT:    ret
56;
57; LA64-LABEL: lshr_i32:
58; LA64:       # %bb.0:
59; LA64-NEXT:    srl.w $a0, $a0, $a1
60; LA64-NEXT:    ret
61  %lshr = lshr i32 %x, %y
62  ret i32 %lshr
63}
64
65define i64 @lshr_i64(i64 %x, i64 %y) {
66; LA32-LABEL: lshr_i64:
67; LA32:       # %bb.0:
68; LA32-NEXT:    srl.w $a0, $a0, $a2
69; LA32-NEXT:    xori $a3, $a2, 31
70; LA32-NEXT:    slli.w $a4, $a1, 1
71; LA32-NEXT:    sll.w $a3, $a4, $a3
72; LA32-NEXT:    or $a0, $a0, $a3
73; LA32-NEXT:    addi.w $a3, $a2, -32
74; LA32-NEXT:    slti $a4, $a3, 0
75; LA32-NEXT:    maskeqz $a0, $a0, $a4
76; LA32-NEXT:    srl.w $a5, $a1, $a3
77; LA32-NEXT:    masknez $a4, $a5, $a4
78; LA32-NEXT:    or $a0, $a0, $a4
79; LA32-NEXT:    srl.w $a1, $a1, $a2
80; LA32-NEXT:    srai.w $a2, $a3, 31
81; LA32-NEXT:    and $a1, $a2, $a1
82; LA32-NEXT:    ret
83;
84; LA64-LABEL: lshr_i64:
85; LA64:       # %bb.0:
86; LA64-NEXT:    srl.d $a0, $a0, $a1
87; LA64-NEXT:    ret
88  %lshr = lshr i64 %x, %y
89  ret i64 %lshr
90}
91
92define i1 @lshr_i1_3(i1 %x) {
93; LA32-LABEL: lshr_i1_3:
94; LA32:       # %bb.0:
95; LA32-NEXT:    ret
96;
97; LA64-LABEL: lshr_i1_3:
98; LA64:       # %bb.0:
99; LA64-NEXT:    ret
100  %lshr = lshr i1 %x, 3
101  ret i1 %lshr
102}
103
104define i8 @lshr_i8_3(i8 %x) {
105; LA32-LABEL: lshr_i8_3:
106; LA32:       # %bb.0:
107; LA32-NEXT:    bstrpick.w $a0, $a0, 7, 3
108; LA32-NEXT:    ret
109;
110; LA64-LABEL: lshr_i8_3:
111; LA64:       # %bb.0:
112; LA64-NEXT:    bstrpick.d $a0, $a0, 7, 3
113; LA64-NEXT:    ret
114  %lshr = lshr i8 %x, 3
115  ret i8 %lshr
116}
117
118define i16 @lshr_i16_3(i16 %x) {
119; LA32-LABEL: lshr_i16_3:
120; LA32:       # %bb.0:
121; LA32-NEXT:    bstrpick.w $a0, $a0, 15, 3
122; LA32-NEXT:    ret
123;
124; LA64-LABEL: lshr_i16_3:
125; LA64:       # %bb.0:
126; LA64-NEXT:    bstrpick.d $a0, $a0, 15, 3
127; LA64-NEXT:    ret
128  %lshr = lshr i16 %x, 3
129  ret i16 %lshr
130}
131
132define i32 @lshr_i32_3(i32 %x) {
133; LA32-LABEL: lshr_i32_3:
134; LA32:       # %bb.0:
135; LA32-NEXT:    srli.w $a0, $a0, 3
136; LA32-NEXT:    ret
137;
138; LA64-LABEL: lshr_i32_3:
139; LA64:       # %bb.0:
140; LA64-NEXT:    bstrpick.d $a0, $a0, 31, 3
141; LA64-NEXT:    ret
142  %lshr = lshr i32 %x, 3
143  ret i32 %lshr
144}
145
146define i64 @lshr_i64_3(i64 %x) {
147; LA32-LABEL: lshr_i64_3:
148; LA32:       # %bb.0:
149; LA32-NEXT:    slli.w $a2, $a1, 29
150; LA32-NEXT:    srli.w $a0, $a0, 3
151; LA32-NEXT:    or $a0, $a0, $a2
152; LA32-NEXT:    srli.w $a1, $a1, 3
153; LA32-NEXT:    ret
154;
155; LA64-LABEL: lshr_i64_3:
156; LA64:       # %bb.0:
157; LA64-NEXT:    srli.d $a0, $a0, 3
158; LA64-NEXT:    ret
159  %lshr = lshr i64 %x, 3
160  ret i64 %lshr
161}
162