xref: /llvm-project/llvm/test/Transforms/InstCombine/consecutive-ptrmask.ll (revision 38fffa630ee80163dc65e759392ad29798905679)
1aa88df85SNoah Goldstein; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 2
2aa88df85SNoah Goldstein; RUN: opt < %s -passes=instcombine -S | FileCheck %s
3aa88df85SNoah Goldstein
4eb86de63SNikita Popovtarget datalayout = "p1:64:64:64:32"
5eb86de63SNikita Popov
6aa88df85SNoah Goldsteindeclare ptr @llvm.ptrmask.p0.i64(ptr, i64)
7eb86de63SNikita Popovdeclare ptr addrspace(1) @llvm.ptrmask.p1.i32(ptr addrspace(1) , i32)
82c5f2b33SNoah Goldsteindeclare <2 x ptr addrspace(1) > @llvm.ptrmask.v2p1.v2i32(<2 x ptr addrspace(1) >, <2 x i32>)
947b7f33bSNikita Popovdeclare <2 x ptr> @llvm.ptrmask.v2p0.v2i64(<2 x ptr>, <2 x i64>)
10aa88df85SNoah Goldsteindeclare void @use.ptr(ptr)
1147b7f33bSNikita Popov
12aa88df85SNoah Goldsteindefine ptr @fold_2x(ptr %p, i64 %m0, i64 %m1) {
13aa88df85SNoah Goldstein; CHECK-LABEL: define ptr @fold_2x
14aa88df85SNoah Goldstein; CHECK-SAME: (ptr [[P:%.*]], i64 [[M0:%.*]], i64 [[M1:%.*]]) {
15edf2e0e0SNoah Goldstein; CHECK-NEXT:    [[TMP1:%.*]] = and i64 [[M1]], [[M0]]
16edf2e0e0SNoah Goldstein; CHECK-NEXT:    [[P1:%.*]] = call ptr @llvm.ptrmask.p0.i64(ptr [[P]], i64 [[TMP1]])
17aa88df85SNoah Goldstein; CHECK-NEXT:    ret ptr [[P1]]
18aa88df85SNoah Goldstein;
19aa88df85SNoah Goldstein  %p0 = call ptr @llvm.ptrmask.p0.i64(ptr %p, i64 %m0)
20aa88df85SNoah Goldstein  %p1 = call ptr @llvm.ptrmask.p0.i64(ptr %p0, i64 %m1)
21aa88df85SNoah Goldstein  ret ptr %p1
22aa88df85SNoah Goldstein}
23aa88df85SNoah Goldstein
24aa88df85SNoah Goldsteindefine ptr @fold_2x_fail_multiuse(ptr %p, i64 %m0, i64 %m1) {
25aa88df85SNoah Goldstein; CHECK-LABEL: define ptr @fold_2x_fail_multiuse
26aa88df85SNoah Goldstein; CHECK-SAME: (ptr [[P:%.*]], i64 [[M0:%.*]], i64 [[M1:%.*]]) {
27aa88df85SNoah Goldstein; CHECK-NEXT:    [[P0:%.*]] = call ptr @llvm.ptrmask.p0.i64(ptr [[P]], i64 [[M0]])
28aa88df85SNoah Goldstein; CHECK-NEXT:    call void @use.ptr(ptr [[P0]])
29aa88df85SNoah Goldstein; CHECK-NEXT:    [[P1:%.*]] = call ptr @llvm.ptrmask.p0.i64(ptr [[P0]], i64 [[M1]])
30aa88df85SNoah Goldstein; CHECK-NEXT:    ret ptr [[P1]]
31aa88df85SNoah Goldstein;
32aa88df85SNoah Goldstein  %p0 = call ptr @llvm.ptrmask.p0.i64(ptr %p, i64 %m0)
33aa88df85SNoah Goldstein  call void @use.ptr(ptr %p0)
34aa88df85SNoah Goldstein  %p1 = call ptr @llvm.ptrmask.p0.i64(ptr %p0, i64 %m1)
35aa88df85SNoah Goldstein  ret ptr %p1
36aa88df85SNoah Goldstein}
37aa88df85SNoah Goldstein
3847b7f33bSNikita Popovdefine <2 x ptr> @fold_2x_vec(<2 x ptr> %p, <2 x i64> %m0, <2 x i64> %m1) {
3947b7f33bSNikita Popov; CHECK-LABEL: define <2 x ptr> @fold_2x_vec
4047b7f33bSNikita Popov; CHECK-SAME: (<2 x ptr> [[P:%.*]], <2 x i64> [[M0:%.*]], <2 x i64> [[M1:%.*]]) {
4147b7f33bSNikita Popov; CHECK-NEXT:    [[TMP1:%.*]] = and <2 x i64> [[M1]], [[M0]]
4247b7f33bSNikita Popov; CHECK-NEXT:    [[P1:%.*]] = call <2 x ptr> @llvm.ptrmask.v2p0.v2i64(<2 x ptr> [[P]], <2 x i64> [[TMP1]])
4347b7f33bSNikita Popov; CHECK-NEXT:    ret <2 x ptr> [[P1]]
4447b7f33bSNikita Popov;
4547b7f33bSNikita Popov  %p0 = call <2 x ptr> @llvm.ptrmask.v2p0.v2i64(<2 x ptr> %p, <2 x i64> %m0)
4647b7f33bSNikita Popov  %p1 = call <2 x ptr> @llvm.ptrmask.v2p0.v2i64(<2 x ptr> %p0, <2 x i64> %m1)
4747b7f33bSNikita Popov  ret <2 x ptr> %p1
4847b7f33bSNikita Popov}
49eb86de63SNikita Popov
50eb86de63SNikita Popovdefine ptr addrspace(1) @fold_2x_smaller_index_type(ptr addrspace(1) %p, i32 %m0, i32 %m1) {
51eb86de63SNikita Popov; CHECK-LABEL: define ptr addrspace(1) @fold_2x_smaller_index_type
52eb86de63SNikita Popov; CHECK-SAME: (ptr addrspace(1) [[P:%.*]], i32 [[M0:%.*]], i32 [[M1:%.*]]) {
53eb86de63SNikita Popov; CHECK-NEXT:    [[TMP1:%.*]] = and i32 [[M1]], [[M0]]
54eb86de63SNikita Popov; CHECK-NEXT:    [[P1:%.*]] = call ptr addrspace(1) @llvm.ptrmask.p1.i32(ptr addrspace(1) [[P]], i32 [[TMP1]])
55eb86de63SNikita Popov; CHECK-NEXT:    ret ptr addrspace(1) [[P1]]
56eb86de63SNikita Popov;
57eb86de63SNikita Popov  %p0 = call ptr addrspace(1) @llvm.ptrmask.p1.i32(ptr addrspace(1) %p, i32 %m0)
58eb86de63SNikita Popov  %p1 = call ptr addrspace(1) @llvm.ptrmask.p1.i32(ptr addrspace(1) %p0, i32 %m1)
59eb86de63SNikita Popov  ret ptr addrspace(1) %p1
60eb86de63SNikita Popov}
612c5f2b33SNoah Goldstein
622c5f2b33SNoah Goldsteindefine <2 x ptr> @fold_2x_vec_i64(<2 x ptr> %p, <2 x i64> %m0) {
632c5f2b33SNoah Goldstein; CHECK-LABEL: define <2 x ptr> @fold_2x_vec_i64
642c5f2b33SNoah Goldstein; CHECK-SAME: (<2 x ptr> [[P:%.*]], <2 x i64> [[M0:%.*]]) {
65*38fffa63SPaul Walker; CHECK-NEXT:    [[TMP1:%.*]] = and <2 x i64> [[M0]], splat (i64 -2)
6651abbf98SNoah Goldstein; CHECK-NEXT:    [[P1:%.*]] = call align 2 <2 x ptr> @llvm.ptrmask.v2p0.v2i64(<2 x ptr> [[P]], <2 x i64> [[TMP1]])
672c5f2b33SNoah Goldstein; CHECK-NEXT:    ret <2 x ptr> [[P1]]
682c5f2b33SNoah Goldstein;
692c5f2b33SNoah Goldstein  %p0 = call <2 x ptr> @llvm.ptrmask.v2p0.v2i64(<2 x ptr> %p, <2 x i64> %m0)
702c5f2b33SNoah Goldstein  %p1 = call <2 x ptr> @llvm.ptrmask.v2p0.v2i64(<2 x ptr> %p0, <2 x i64> <i64 -2, i64 -2>)
712c5f2b33SNoah Goldstein  ret <2 x ptr> %p1
722c5f2b33SNoah Goldstein}
732c5f2b33SNoah Goldstein
742c5f2b33SNoah Goldsteindefine <2 x ptr addrspace(1) > @fold_2x_vec_i32_undef(<2 x ptr addrspace(1) > %p, <2 x i32> %m0) {
752c5f2b33SNoah Goldstein; CHECK-LABEL: define <2 x ptr addrspace(1)> @fold_2x_vec_i32_undef
762c5f2b33SNoah Goldstein; CHECK-SAME: (<2 x ptr addrspace(1)> [[P:%.*]], <2 x i32> [[M0:%.*]]) {
772c5f2b33SNoah Goldstein; CHECK-NEXT:    [[TMP1:%.*]] = and <2 x i32> [[M0]], <i32 -2, i32 undef>
782c5f2b33SNoah Goldstein; CHECK-NEXT:    [[P1:%.*]] = call <2 x ptr addrspace(1)> @llvm.ptrmask.v2p1.v2i32(<2 x ptr addrspace(1)> [[P]], <2 x i32> [[TMP1]])
792c5f2b33SNoah Goldstein; CHECK-NEXT:    ret <2 x ptr addrspace(1)> [[P1]]
802c5f2b33SNoah Goldstein;
812c5f2b33SNoah Goldstein  %p0 = call <2 x ptr addrspace(1) > @llvm.ptrmask.v2p1.v2i32(<2 x ptr addrspace(1) > %p, <2 x i32> %m0)
822c5f2b33SNoah Goldstein  %p1 = call <2 x ptr addrspace(1) > @llvm.ptrmask.v2p1.v2i32(<2 x ptr addrspace(1) > %p0, <2 x i32> <i32 -2, i32 undef>)
832c5f2b33SNoah Goldstein  ret <2 x ptr addrspace(1) > %p1
842c5f2b33SNoah Goldstein}
85