[llvm] d6a7da5 - MemoryBuiltins: only claim an allocator family on builtin functions

[Augie Fackler via llvm-commits]

Author: Augie Fackler
Date: 2022-04-07T12:38:45-04:00
New Revision: d6a7da5ae3a2b2bef395a60af10623aa958d97e6

URL: https://github.com/llvm/llvm-project/commit/d6a7da5ae3a2b2bef395a60af10623aa958d97e6
DIFF: https://github.com/llvm/llvm-project/commit/d6a7da5ae3a2b2bef395a60af10623aa958d97e6.diff

LOG: MemoryBuiltins: only claim an allocator family on builtin functions

This lines up with other parts of the codebase that only use special
knowledge about allocator functions if they're builtins.

Differential Revision: https://reviews.llvm.org/D123053

Added: 
    

Modified: 
    llvm/lib/Analysis/MemoryBuiltins.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Analysis/MemoryBuiltins.cpp b/llvm/lib/Analysis/MemoryBuiltins.cpp
index 12edd4607b977..4db57957453b5 100644
--- a/llvm/lib/Analysis/MemoryBuiltins.cpp
+++ b/llvm/lib/Analysis/MemoryBuiltins.cpp
@@ -492,7 +492,7 @@ Optional<StringRef> llvm::getAllocationFamily(const Value *I,
                                               const TargetLibraryInfo *TLI) {
   bool IsNoBuiltin;
   const Function *Callee = getCalledFunction(I, IsNoBuiltin);
-  if (Callee == nullptr)
+  if (Callee == nullptr || IsNoBuiltin)
     return None;
   LibFunc TLIFn;
   if (!TLI || !TLI->getLibFunc(*Callee, TLIFn) || !TLI->has(TLIFn))