[AArch64][NFC] NFC for const vector as Instruction operand (#116790)
[llvm-project.git] / llvm / test / Transforms / InstCombine / consecutive-ptrmask.ll
blob3e5468d2f8bed5e667cfc71398833ff62a33c9a2
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
4 target datalayout = "p1:64:64:64:32"
6 declare ptr @llvm.ptrmask.p0.i64(ptr, i64)
7 declare ptr addrspace(1) @llvm.ptrmask.p1.i32(ptr addrspace(1) , i32)
8 declare <2 x ptr addrspace(1) > @llvm.ptrmask.v2p1.v2i32(<2 x ptr addrspace(1) >, <2 x i32>)
9 declare <2 x ptr> @llvm.ptrmask.v2p0.v2i64(<2 x ptr>, <2 x i64>)
10 declare void @use.ptr(ptr)
12 define 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]]
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
24 define 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]]
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
38 define <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]]
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
50 define 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]]
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
62 define <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]]
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
74 define <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]]
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