diff options
-rw-r--r-- | llvm/lib/Analysis/Loads.cpp | 5 | ||||
-rw-r--r-- | llvm/test/Analysis/ValueTracking/memory-dereferenceable.ll | 5 |
2 files changed, 9 insertions, 1 deletions
diff --git a/llvm/lib/Analysis/Loads.cpp b/llvm/lib/Analysis/Loads.cpp index 309a1d95efb..75426b54195 100644 --- a/llvm/lib/Analysis/Loads.cpp +++ b/llvm/lib/Analysis/Loads.cpp @@ -101,6 +101,11 @@ static bool isDereferenceableAndAlignedPointer( return isDereferenceableAndAlignedPointer(ASC->getOperand(0), Align, Size, DL, CtxI, DT, Visited); + if (auto CS = ImmutableCallSite(V)) + if (const Value *RV = CS.getReturnedArgOperand()) + return isDereferenceableAndAlignedPointer(RV, Align, Size, DL, CtxI, DT, + Visited); + // If we don't know, assume the worst. return false; } diff --git a/llvm/test/Analysis/ValueTracking/memory-dereferenceable.ll b/llvm/test/Analysis/ValueTracking/memory-dereferenceable.ll index 15a93328974..29c31b95b18 100644 --- a/llvm/test/Analysis/ValueTracking/memory-dereferenceable.ll +++ b/llvm/test/Analysis/ValueTracking/memory-dereferenceable.ll @@ -43,7 +43,8 @@ entry: %load4 = load i32, i32 addrspace(1)* %relocate ; CHECK-NOT: %nparam - %nparam = getelementptr i32, i32 addrspace(1)* %dparam, i32 5 + %dpa = call i32 addrspace(1)* @func1(i32 addrspace(1)* %dparam) + %nparam = getelementptr i32, i32 addrspace(1)* %dpa, i32 5 %load5 = load i32, i32 addrspace(1)* %nparam ; Load from a non-dereferenceable load @@ -162,6 +163,8 @@ if.end: declare token @llvm.experimental.gc.statepoint.p0f_i1f(i64, i32, i1 ()*, i32, i32, ...) declare i32 addrspace(1)* @llvm.experimental.gc.relocate.p1i32(token, i32, i32) +declare i32 addrspace(1)* @func1(i32 addrspace(1)* returned) nounwind argmemonly + !0 = !{i64 4} !1 = !{i64 2} !2 = !{} |