[PATCH] D117242: [Attributor] Generalize heap to stack to any allocator with relevant properties
Philip Reames via Phabricator via llvm-commits
llvm-commits at lists.llvm.org
Thu Jan 13 12:22:22 PST 2022
reames created this revision.
reames added reviewers: Bryce-MW, durin42, jdoerfert.
Herald added subscribers: ormris, okura, kuter, uenoku, bollu, hiraditya, mcrosier.
Herald added a reviewer: uenoku.
Herald added a reviewer: homerdin.
reames requested review of this revision.
Herald added a reviewer: sstefan1.
Herald added a reviewer: baziotis.
Herald added a project: LLVM.
This completes removal of the isXLike queries, and depends on a whole series of earlier patches.
Repository:
rG LLVM Github Monorepo
https://reviews.llvm.org/D117242
Files:
llvm/lib/Transforms/IPO/AttributorAttributes.cpp
Index: llvm/lib/Transforms/IPO/AttributorAttributes.cpp
===================================================================
--- llvm/lib/Transforms/IPO/AttributorAttributes.cpp
+++ llvm/lib/Transforms/IPO/AttributorAttributes.cpp
@@ -5771,13 +5771,6 @@
/// The call that allocates the memory.
CallBase *const CB;
- /// The kind of allocation.
- const enum class AllocationKind {
- MALLOC,
- CALLOC,
- ALIGNED_ALLOC,
- } Kind;
-
/// The library function id for the allocation.
LibFunc LibraryFunctionId = NotLibFunc;
@@ -5834,20 +5827,17 @@
DeallocationInfos[CB] = new (A.Allocator) DeallocationInfo{CB};
return true;
}
- bool IsMalloc = isMallocLikeFn(CB, TLI);
- bool IsAlignedAllocLike = !IsMalloc && isAlignedAllocLikeFn(CB, TLI);
- bool IsCalloc =
- !IsMalloc && !IsAlignedAllocLike && isCallocLikeFn(CB, TLI);
- if (!IsMalloc && !IsAlignedAllocLike && !IsCalloc)
- return true;
- auto Kind =
- IsMalloc ? AllocationInfo::AllocationKind::MALLOC
- : (IsCalloc ? AllocationInfo::AllocationKind::CALLOC
- : AllocationInfo::AllocationKind::ALIGNED_ALLOC);
-
- AllocationInfo *AI = new (A.Allocator) AllocationInfo{CB, Kind};
- AllocationInfos[CB] = AI;
- TLI->getLibFunc(*CB, AI->LibraryFunctionId);
+ // To do heap to stack, we need to know that the allocation itself is
+ // removable once uses are rewritten, and that we can initialize the
+ // alloca to the same pattern as the original allocation result.
+ if (isAllocationFn(CB, TLI) && isAllocRemovable(CB, TLI)) {
+ auto *I8Ty = Type::getInt8Ty(CB->getParent()->getContext());
+ if (nullptr != getInitialValueOfAllocation(CB, TLI, I8Ty)) {
+ AllocationInfo *AI = new (A.Allocator) AllocationInfo{CB};
+ AllocationInfos[CB] = AI;
+ TLI->getLibFunc(*CB, AI->LibraryFunctionId);
+ }
+ }
return true;
};
-------------- next part --------------
A non-text attachment was scrubbed...
Name: D117242.399751.patch
Type: text/x-patch
Size: 2031 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20220113/b0fd6abd/attachment.bin>
More information about the llvm-commits
mailing list