[AArch64] Fix bug in prolog clobbering live reg when shrink wrapping.
[oota-llvm.git] / test / CodeGen / AArch64 / or-combine.ll
1 ; RUN: llc -mtriple=aarch64-linux-gnu -o - %s | FileCheck %s
2
3 define i32 @test_consts(i32 %in) {
4 ; CHECK-LABEL: test_consts:
5 ; CHECK-NOT: bfxil
6 ; CHECK-NOT: and
7 ; CHECK-NOT: orr
8 ; CHECK: ret
9
10   %lo = and i32 %in, 65535
11   %hi = and i32 %in, -65536
12   %res = or i32 %lo, %hi
13   ret i32 %res
14 }
15
16 define i32 @test_generic(i32 %in, i32 %mask1, i32 %mask2) {
17 ; CHECK-LABEL: test_generic:
18 ; CHECK: orr [[FULL_MASK:w[0-9]+]], w1, w2
19 ; CHECK: and w0, w0, [[FULL_MASK]]
20
21   %lo = and i32 %in, %mask1
22   %hi = and i32 %in, %mask2
23   %res = or i32 %lo, %hi
24   ret i32 %res
25 }
26
27 ; In this case the transformation isn't profitable, since %lo and %hi
28 ; are used more than once.
29 define [3 x i32] @test_reuse(i32 %in, i32 %mask1, i32 %mask2) {
30 ; CHECK-LABEL: test_reuse:
31 ; CHECK-DAG: and w1, w0, w1
32 ; CHECK-DAG: and w2, w0, w2
33 ; CHECK-DAG: orr w0, w1, w2
34
35   %lo = and i32 %in, %mask1
36   %hi = and i32 %in, %mask2
37   %recombine = or i32 %lo, %hi
38
39   %res.tmp0 = insertvalue [3 x i32] undef, i32 %recombine, 0
40   %res.tmp1 = insertvalue [3 x i32] %res.tmp0, i32 %lo, 1
41   %res = insertvalue [3 x i32] %res.tmp1, i32 %hi, 2
42
43   ret [3 x i32] %res
44 }