xref: /llvm-project/llvm/test/Transforms/InstCombine/consecutive-ptrmask.ll (revision 38fffa630ee80163dc65e759392ad29798905679) 
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 2
2; RUN: opt < %s -passes=instcombine -S | FileCheck %s
3
4target datalayout = "p1:64:64:64:32"
5
6declare ptr @llvm.ptrmask.p0.i64(ptr, i64)
7declare ptr addrspace(1) @llvm.ptrmask.p1.i32(ptr addrspace(1) , i32)
8declare <2 x ptr addrspace(1) > @llvm.ptrmask.v2p1.v2i32(<2 x ptr addrspace(1) >, <2 x i32>)
9declare <2 x ptr> @llvm.ptrmask.v2p0.v2i64(<2 x ptr>, <2 x i64>)
10declare void @use.ptr(ptr)
11
12define ptr @fold_2x(ptr %p, i64 %m0, i64 %m1) {
13; CHECK-LABEL: define ptr @fold_2x
14; CHECK-SAME: (ptr [[P:%.*]], i64 [[M0:%.*]], i64 [[M1:%.*]]) {
15; CHECK-NEXT:    [[TMP1:%.*]] = and i64 [[M1]], [[M0]]
16; CHECK-NEXT:    [[P1:%.*]] = call ptr @llvm.ptrmask.p0.i64(ptr [[P]], i64 [[TMP1]])
17; CHECK-NEXT:    ret ptr [[P1]]
18;
19  %p0 = call ptr @llvm.ptrmask.p0.i64(ptr %p, i64 %m0)
20  %p1 = call ptr @llvm.ptrmask.p0.i64(ptr %p0, i64 %m1)
21  ret ptr %p1
22}
23
24define ptr @fold_2x_fail_multiuse(ptr %p, i64 %m0, i64 %m1) {
25; CHECK-LABEL: define ptr @fold_2x_fail_multiuse
26; CHECK-SAME: (ptr [[P:%.*]], i64 [[M0:%.*]], i64 [[M1:%.*]]) {
27; CHECK-NEXT:    [[P0:%.*]] = call ptr @llvm.ptrmask.p0.i64(ptr [[P]], i64 [[M0]])
28; CHECK-NEXT:    call void @use.ptr(ptr [[P0]])
29; CHECK-NEXT:    [[P1:%.*]] = call ptr @llvm.ptrmask.p0.i64(ptr [[P0]], i64 [[M1]])
30; CHECK-NEXT:    ret ptr [[P1]]
31;
32  %p0 = call ptr @llvm.ptrmask.p0.i64(ptr %p, i64 %m0)
33  call void @use.ptr(ptr %p0)
34  %p1 = call ptr @llvm.ptrmask.p0.i64(ptr %p0, i64 %m1)
35  ret ptr %p1
36}
37
38define <2 x ptr> @fold_2x_vec(<2 x ptr> %p, <2 x i64> %m0, <2 x i64> %m1) {
39; CHECK-LABEL: define <2 x ptr> @fold_2x_vec
40; CHECK-SAME: (<2 x ptr> [[P:%.*]], <2 x i64> [[M0:%.*]], <2 x i64> [[M1:%.*]]) {
41; CHECK-NEXT:    [[TMP1:%.*]] = and <2 x i64> [[M1]], [[M0]]
42; CHECK-NEXT:    [[P1:%.*]] = call <2 x ptr> @llvm.ptrmask.v2p0.v2i64(<2 x ptr> [[P]], <2 x i64> [[TMP1]])
43; CHECK-NEXT:    ret <2 x ptr> [[P1]]
44;
45  %p0 = call <2 x ptr> @llvm.ptrmask.v2p0.v2i64(<2 x ptr> %p, <2 x i64> %m0)
46  %p1 = call <2 x ptr> @llvm.ptrmask.v2p0.v2i64(<2 x ptr> %p0, <2 x i64> %m1)
47  ret <2 x ptr> %p1
48}
49
50define ptr addrspace(1) @fold_2x_smaller_index_type(ptr addrspace(1) %p, i32 %m0, i32 %m1) {
51; CHECK-LABEL: define ptr addrspace(1) @fold_2x_smaller_index_type
52; CHECK-SAME: (ptr addrspace(1) [[P:%.*]], i32 [[M0:%.*]], i32 [[M1:%.*]]) {
53; CHECK-NEXT:    [[TMP1:%.*]] = and i32 [[M1]], [[M0]]
54; CHECK-NEXT:    [[P1:%.*]] = call ptr addrspace(1) @llvm.ptrmask.p1.i32(ptr addrspace(1) [[P]], i32 [[TMP1]])
55; CHECK-NEXT:    ret ptr addrspace(1) [[P1]]
56;
57  %p0 = call ptr addrspace(1) @llvm.ptrmask.p1.i32(ptr addrspace(1) %p, i32 %m0)
58  %p1 = call ptr addrspace(1) @llvm.ptrmask.p1.i32(ptr addrspace(1) %p0, i32 %m1)
59  ret ptr addrspace(1) %p1
60}
61
62define <2 x ptr> @fold_2x_vec_i64(<2 x ptr> %p, <2 x i64> %m0) {
63; CHECK-LABEL: define <2 x ptr> @fold_2x_vec_i64
64; CHECK-SAME: (<2 x ptr> [[P:%.*]], <2 x i64> [[M0:%.*]]) {
65; CHECK-NEXT:    [[TMP1:%.*]] = and <2 x i64> [[M0]], splat (i64 -2)
66; CHECK-NEXT:    [[P1:%.*]] = call align 2 <2 x ptr> @llvm.ptrmask.v2p0.v2i64(<2 x ptr> [[P]], <2 x i64> [[TMP1]])
67; CHECK-NEXT:    ret <2 x ptr> [[P1]]
68;
69  %p0 = call <2 x ptr> @llvm.ptrmask.v2p0.v2i64(<2 x ptr> %p, <2 x i64> %m0)
70  %p1 = call <2 x ptr> @llvm.ptrmask.v2p0.v2i64(<2 x ptr> %p0, <2 x i64> <i64 -2, i64 -2>)
71  ret <2 x ptr> %p1
72}
73
74define <2 x ptr addrspace(1) > @fold_2x_vec_i32_undef(<2 x ptr addrspace(1) > %p, <2 x i32> %m0) {
75; CHECK-LABEL: define <2 x ptr addrspace(1)> @fold_2x_vec_i32_undef
76; CHECK-SAME: (<2 x ptr addrspace(1)> [[P:%.*]], <2 x i32> [[M0:%.*]]) {
77; CHECK-NEXT:    [[TMP1:%.*]] = and <2 x i32> [[M0]], <i32 -2, i32 undef>
78; CHECK-NEXT:    [[P1:%.*]] = call <2 x ptr addrspace(1)> @llvm.ptrmask.v2p1.v2i32(<2 x ptr addrspace(1)> [[P]], <2 x i32> [[TMP1]])
79; CHECK-NEXT:    ret <2 x ptr addrspace(1)> [[P1]]
80;
81  %p0 = call <2 x ptr addrspace(1) > @llvm.ptrmask.v2p1.v2i32(<2 x ptr addrspace(1) > %p, <2 x i32> %m0)
82  %p1 = call <2 x ptr addrspace(1) > @llvm.ptrmask.v2p1.v2i32(<2 x ptr addrspace(1) > %p0, <2 x i32> <i32 -2, i32 undef>)
83  ret <2 x ptr addrspace(1) > %p1
84}
85