Add some jemalloc functions to TargetLibraryInfo + MemoryBuiltins

Original patch by Nathaniel Theis

Author: alexcrichton

include/llvm/Analysis/TargetLibraryInfo.def

@@ -550,6 +550,12 @@ TLI_DEFINE_STRING_INTERNAL("isascii")
 /// int isdigit(int c);
 TLI_DEFINE_ENUM_INTERNAL(isdigit)
 TLI_DEFINE_STRING_INTERNAL("isdigit")
+/// void *je_mallocx(size_t size, int flags)
+TLI_DEFINE_ENUM_INTERNAL(je_mallocx)
+TLI_DEFINE_STRING_INTERNAL("je_mallocx")
+/// void je_sdallocx(void *ptr, size_t size, int flags)
+TLI_DEFINE_ENUM_INTERNAL(je_sdallocx)
+TLI_DEFINE_STRING_INTERNAL("je_sdallocx")
 /// long int labs(long int j);
 TLI_DEFINE_ENUM_INTERNAL(labs)
 TLI_DEFINE_STRING_INTERNAL("labs")

lib/Analysis/MemoryBuiltins.cpp

@@ -66,7 +66,8 @@ static const AllocFnsTy AllocationFnData[] = {
   {LibFunc::realloc,             ReallocLike, 2, 1,  -1},
   {LibFunc::reallocf,            ReallocLike, 2, 1,  -1},
   {LibFunc::strdup,              StrDupLike,  1, -1, -1},
-  {LibFunc::strndup,             StrDupLike,  2, 1,  -1}
+  {LibFunc::strndup,             StrDupLike,  2, 1,  -1},
+  {LibFunc::je_mallocx,          MallocLike,  2, 0,  -1}
   // TODO: Handle "int posix_memalign(void **, size_t, size_t)"
 };
 
@@ -322,6 +323,8 @@ const CallInst *llvm::isFreeCall(const Value *I, const TargetLibraryInfo *TLI) {
            TLIFn == LibFunc::ZdaPvm ||              // delete[](void*, ulong)
            TLIFn == LibFunc::ZdaPvRKSt9nothrow_t)   // delete[](void*, nothrow)
     ExpectedNumParams = 2;
+  else if (TLIFn == LibFunc::je_sdallocx)
+    ExpectedNumParams = 3;
   else
     return nullptr;