diff --git a/llvm/lib/IR/Instructions.cpp b/llvm/lib/IR/Instructions.cpp --- a/llvm/lib/IR/Instructions.cpp +++ b/llvm/lib/IR/Instructions.cpp @@ -45,6 +45,10 @@ using namespace llvm; +static cl::opt DisableI2pP2iOpt( + "disable-i2p-p2i-opt", cl::init(false), + cl::desc("Disables inttoptr/ptrtoint roundtrip optimization")); + //===----------------------------------------------------------------------===// // AllocaInst Class //===----------------------------------------------------------------------===// @@ -2841,6 +2845,9 @@ return secondOp; return 0; case 7: { + // Disable inttoptr/ptrtoint optimization if enabled. + if (DisableI2pP2iOpt) + return 0; // Cannot simplify if address spaces are different! if (SrcTy->getPointerAddressSpace() != DstTy->getPointerAddressSpace()) return 0; diff --git a/llvm/test/Transforms/InstCombine/ptr-int-ptr-icmp.ll b/llvm/test/Transforms/InstCombine/ptr-int-ptr-icmp.ll new file mode 100644 --- /dev/null +++ b/llvm/test/Transforms/InstCombine/ptr-int-ptr-icmp.ll @@ -0,0 +1,93 @@ +; NOTE: Assertions have been autogenerated by utils/update_test_checks.py +; RUN: opt -instcombine -S -disable-i2p-p2i-opt < %s | FileCheck %s + +target datalayout = "e-p:64:64-p1:16:16-p2:32:32:32-p3:64:64:64" +target triple = "x86_64-unknown-linux-gnu" + +; icmp (inttoptr (ptrtoint p1)), p2 --> icmp p1, p2. + +define i1 @func(i8* %X, i8* %Y) { +; CHECK-LABEL: @func( +; CHECK-NEXT: [[I:%.*]] = ptrtoint i8* [[X:%.*]] to i64 +; CHECK-NEXT: [[P:%.*]] = inttoptr i64 [[I]] to i8* +; CHECK-NEXT: [[CMP:%.*]] = icmp eq i8* [[P]], [[Y:%.*]] +; CHECK-NEXT: ret i1 [[CMP]] +; + %i = ptrtoint i8* %X to i64 + %p = inttoptr i64 %i to i8* + %cmp = icmp eq i8* %p, %Y + ret i1 %cmp +} + +define i1 @func2(i16* %X, i8* %Y) { +; CHECK-LABEL: @func2( +; CHECK-NEXT: [[I:%.*]] = ptrtoint i16* [[X:%.*]] to i64 +; CHECK-NEXT: [[P:%.*]] = inttoptr i64 [[I]] to i8* +; CHECK-NEXT: [[CMP:%.*]] = icmp eq i8* [[P]], [[Y:%.*]] +; CHECK-NEXT: ret i1 [[CMP]] +; + %i = ptrtoint i16* %X to i64 + %p = inttoptr i64 %i to i8* + %cmp = icmp eq i8* %p, %Y + ret i1 %cmp +} + +define i1 @func3(i16* %X, i8* %Y) { +; CHECK-LABEL: @func3( +; CHECK-NEXT: [[I:%.*]] = ptrtoint i16* [[X:%.*]] to i64 +; CHECK-NEXT: [[P:%.*]] = inttoptr i64 [[I]] to i8* +; CHECK-NEXT: [[CMP:%.*]] = icmp eq i8* [[P]], [[Y:%.*]] +; CHECK-NEXT: ret i1 [[CMP]] +; + %i = ptrtoint i16* %X to i64 + %p = inttoptr i64 %i to i8* + %cmp = icmp eq i8* %Y, %p + ret i1 %cmp +} + +; Negative test - Wrong Integer type. + +define i1 @func4(i16* %X, i8* %Y) { +; CHECK-LABEL: @func4( +; CHECK-NEXT: [[TMP1:%.*]] = ptrtoint i16* [[X:%.*]] to i64 +; CHECK-NEXT: [[TMP2:%.*]] = and i64 [[TMP1]], 4294967295 +; CHECK-NEXT: [[P:%.*]] = inttoptr i64 [[TMP2]] to i8* +; CHECK-NEXT: [[CMP:%.*]] = icmp eq i8* [[P]], [[Y:%.*]] +; CHECK-NEXT: ret i1 [[CMP]] +; + %i = ptrtoint i16* %X to i32 + %p = inttoptr i32 %i to i8* + %cmp = icmp eq i8* %Y, %p + ret i1 %cmp +} + +; Negative test - Pointers in different address space + +define i1 @func5(i8* %X, i16 addrspace(3)* %Y){ +; CHECK-LABEL: @func5( +; CHECK-NEXT: [[I:%.*]] = ptrtoint i8* [[X:%.*]] to i64 +; CHECK-NEXT: [[P:%.*]] = inttoptr i64 [[I]] to i16 addrspace(3)* +; CHECK-NEXT: [[CMP:%.*]] = icmp eq i16 addrspace(3)* [[P]], [[Y:%.*]] +; CHECK-NEXT: ret i1 [[CMP]] +; + %i = ptrtoint i8* %X to i64 + %p = inttoptr i64 %i to i16 addrspace(3)* + %cmp = icmp eq i16 addrspace(3)* %Y, %p + ret i1 %cmp +} + +; Negative test - Pointers in different address space + +define i1 @func6(i8 addrspace(2)* %X, i16* %Y){ +; CHECK-LABEL: @func6( +; CHECK-NEXT: [[TMP1:%.*]] = ptrtoint i8 addrspace(2)* [[X:%.*]] to i32 +; CHECK-NEXT: [[I:%.*]] = zext i32 [[TMP1]] to i64 +; CHECK-NEXT: [[P:%.*]] = inttoptr i64 [[I]] to i16* +; CHECK-NEXT: [[CMP:%.*]] = icmp eq i16* [[P]], [[Y:%.*]] +; CHECK-NEXT: ret i1 [[CMP]] +; + %i = ptrtoint i8 addrspace(2)* %X to i64 + %p = inttoptr i64 %i to i16* + %cmp = icmp eq i16* %Y, %p + ret i1 %cmp +}