diff --git a/llvm/lib/IR/Value.cpp b/llvm/lib/IR/Value.cpp --- a/llvm/lib/IR/Value.cpp +++ b/llvm/lib/IR/Value.cpp @@ -728,6 +728,64 @@ return stripPointerCastsAndOffsets(this, Func); } +// Return true if the memory object referred to by V can by freed in the scope +// for which the SSA value defining the allocation is statically defined. E.g. +// deallocation after the static scope of a value does not count. +static bool canBeFreed(const Value *V) { + assert(V->getType()->isPointerTy()); + + // Cases that can simply never be deallocated + // *) Constants aren't allocated per se, thus not deallocated either. + if (isa(V)) + return false; + + const Function *F = nullptr; + if (auto *I = dyn_cast(V)) + F = I->getFunction(); + if (auto *A = dyn_cast(V)) + F = A->getParent(); + + if (!F) + return true; + + // A pointer to an object in a function which neither frees, nor can arrange + // for another thread to free on its behalf, can not be freed in the scope + // of the function. + if (F->doesNotFreeMemory() && F->hasNoSync()) + return false; + + // With garbage collection, deallocation typically occurs solely at or after + // safepoints. If we're compiling for a collector which uses the + // gc.statepoint infrastructure, safepoints aren't explicitly present + // in the IR until after lowering from abstract to physical machine model. + // The collector could chose to mix explicit deallocation and gc'd objects + // which is why we need the explicit opt in on a per collector basis. + if (!F->hasGC()) + return true; + + const auto &GCName = F->getGC(); + const StringRef StatepointExampleName("statepoint-example"); + if (GCName != StatepointExampleName) + return true; + + auto *PT = cast(V->getType()); + if (PT->getAddressSpace() != 1) + // For the sake of this example GC, we arbitrarily pick addrspace(1) as our + // GC managed heap. This must match the same check in + // RewriteStatepointsForGC (and probably needs better factored.) + return true; + + // It is cheaper to scan for a declaration than to scan for a use in this + // function. Note that gc.statepoint is a type overloaded function so the + // usual trick of requesting declaration of the intrinsic from the module + // doesn't work. + for (auto &Fn : *F->getParent()) + if (Fn.getIntrinsicID() == Intrinsic::experimental_gc_statepoint) + return true; + return false; +} + + uint64_t Value::getPointerDereferenceableBytes(const DataLayout &DL, bool &CanBeNull, bool &CanBeFreed) const { @@ -735,7 +793,7 @@ uint64_t DerefBytes = 0; CanBeNull = false; - CanBeFreed = UseDerefAtPointSemantics; + CanBeFreed = UseDerefAtPointSemantics && canBeFreed(this); if (const Argument *A = dyn_cast(this)) { DerefBytes = A->getDereferenceableBytes(); if (DerefBytes == 0) { @@ -798,7 +856,6 @@ // CanBeNull flag. DerefBytes = DL.getTypeStoreSize(GV->getValueType()).getFixedSize(); CanBeNull = false; - CanBeFreed = false; } } return DerefBytes; diff --git a/llvm/test/Analysis/ValueTracking/deref-abstract-gc.ll b/llvm/test/Analysis/ValueTracking/deref-abstract-gc.ll --- a/llvm/test/Analysis/ValueTracking/deref-abstract-gc.ll +++ b/llvm/test/Analysis/ValueTracking/deref-abstract-gc.ll @@ -7,13 +7,12 @@ ; conceptually live forever. But there may be non-managed objects which are ; freed. ; CHECK-LABEL: 'abstract_model' -; CHECK-NOT: %gc_ptr +; CHECK: %gc_ptr ; CHECK-NOT: %other_ptr ; FIXME: Can infer the gc pointer case define void @abstract_model(i32 addrspace(1)* dereferenceable(8) %gc_ptr, i32* dereferenceable(8) %other_ptr) gc "statepoint-example" { -; CHECK: The following are dereferenceable: entry: call void @mayfree() load i32, i32 addrspace(1)* %gc_ptr diff --git a/llvm/test/Analysis/ValueTracking/memory-dereferenceable.ll b/llvm/test/Analysis/ValueTracking/memory-dereferenceable.ll --- a/llvm/test/Analysis/ValueTracking/memory-dereferenceable.ll +++ b/llvm/test/Analysis/ValueTracking/memory-dereferenceable.ll @@ -245,16 +245,14 @@ ; CHECK-LABEL: 'negative' ; GLOBAL: %p ; POINT-NOT: %p -define void @negative(i32* dereferenceable(8) %p) nofree nosync { +define void @negative(i32* dereferenceable(8) %p) { call void @mayfree() %v = load i32, i32* %p ret void } ; CHECK-LABEL: 'infer_func_attrs1' -; GLOBAL: %p -; POINT-NOT: %p -; FIXME: Can be inferred from attributes +; CHECK: %p define void @infer_func_attrs1(i32* dereferenceable(8) %p) nofree nosync { call void @mayfree() %v = load i32, i32* %p