[llvm] [memprof] Teach extractCallsFromIR to recognize heap allocation functions (PR #115938)

Snehasish Kumar via llvm-commits llvm-commits at lists.llvm.org
Tue Nov 12 13:57:35 PST 2024


================
@@ -795,8 +795,8 @@ struct AllocMatchInfo {
   bool Matched = false;
 };
 
-DenseMap<uint64_t, SmallVector<CallEdgeTy, 0>>
-memprof::extractCallsFromIR(Module &M) {
+DenseMap<uint64_t, SmallVector<CallEdgeTy, 0>> memprof::extractCallsFromIR(
+    Module &M, function_ref<const TargetLibraryInfo &(Function &)> GetTLI) {
----------------
snehasish wrote:

I think you can use the WrapperPass based approach on L117 here: https://github.com/llvm/llvm-project/commit/2e426fe8ff314c2565073e73e27fdbdf36c140a3#diff-b13f32dcb9d84dd5b7c4d5964bad73df362fbbe51ac3e551765aca1e9299c2b6R117



https://github.com/llvm/llvm-project/pull/115938


More information about the llvm-commits mailing list