[llvm] [SampleFDO] Match functions with the same base function name (PR #126688)

[Lei Wang via llvm-commits]

================
@@ -727,6 +728,33 @@ bool SampleProfileMatcher::functionMatchesProfileHelper(
   // two sequences are.
   float Similarity = 0.0;
 
+  // Match the functions if they have the same base name(after demangling) and
+  // skip the similarity check.
+  ItaniumPartialDemangler Demangler;
+  // Helper lambda to demangle and get the base name. If the demangling failed,
+  // return an empty string.
+  auto GetBaseName = [&](StringRef FName) {
+    auto FunctionName = FName.str();
+    if (Demangler.partialDemangle(FunctionName.c_str()))
+      return std::string();
+    constexpr size_t MaxBaseNameSize = 4096;
+    char BaseNameBuf[MaxBaseNameSize] = {};
+    size_t BaseNameSize = MaxBaseNameSize;
+    char *BaseNamePtr =
+        Demangler.getFunctionBaseName(BaseNameBuf, &BaseNameSize);
+    return (BaseNamePtr && BaseNameSize)
+               ? std::string(BaseNamePtr, BaseNameSize)
+               : std::string();
+  };
+  auto IRBaseName = GetBaseName(IRFunc.getName());
----------------
wlei-llvm wrote:

> I guess if we are not doing the function/profile comparison recursively, what I suggested above won't work.
> Are we doing it recursively when we cannot decide if where a callee funciton/profile is a match?

Yeah, we don't do it recursively, we only look up for two levels for a matched caller.

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