[llvm] [MemProf] Use radix tree for alloc contexts in bitcode summaries (PR #117066)

[LLVM Continuous Integration via llvm-commits]

llvm-ci wrote:

LLVM Buildbot has detected a new failure on builder `mlir-nvidia` running on `mlir-nvidia` while building `llvm` at step 5 "build-check-mlir-build-only".

Full details are available at: https://lab.llvm.org/buildbot/#/builders/138/builds/6846

<details>
<summary>Here is the relevant piece of the build log for the reference</summary>

```
Step 5 (build-check-mlir-build-only) failure: build (failure)
...
36.391 [421/1/4599] Creating library symlink lib/libLLVMSymbolize.so
36.511 [420/1/4600] Linking CXX shared library lib/libLLVMProfileData.so.20.0git
36.517 [419/1/4601] Creating library symlink lib/libLLVMProfileData.so
36.690 [418/1/4602] Linking CXX shared library lib/libLLVMAnalysis.so.20.0git
36.696 [417/1/4603] Creating library symlink lib/libLLVMAnalysis.so
36.782 [412/5/4604] Linking CXX shared library lib/libLLVMIRPrinter.so.20.0git
36.784 [411/5/4605] Linking CXX shared library lib/libLLVMTarget.so.20.0git
36.789 [410/5/4606] Creating library symlink lib/libLLVMIRPrinter.so
36.792 [410/4/4607] Creating library symlink lib/libLLVMTarget.so
36.808 [409/4/4608] Linking CXX shared library lib/libLLVMBitWriter.so.20.0git
FAILED: lib/libLLVMBitWriter.so.20.0git 
: && /usr/bin/clang++ -fPIC -fPIC -fno-semantic-interposition -fvisibility-inlines-hidden -Werror=date-time -Werror=unguarded-availability-new -Wall -Wextra -Wno-unused-parameter -Wwrite-strings -Wcast-qual -Wmissing-field-initializers -pedantic -Wno-long-long -Wc++98-compat-extra-semi -Wimplicit-fallthrough -Wcovered-switch-default -Wno-noexcept-type -Wnon-virtual-dtor -Wdelete-non-virtual-dtor -Wsuggest-override -Wstring-conversion -Wmisleading-indentation -Wctad-maybe-unsupported -fdiagnostics-color -ffunction-sections -fdata-sections -O3 -DNDEBUG  -Wl,-z,defs -Wl,-z,nodelete -fuse-ld=lld -Wl,--color-diagnostics   -Wl,--gc-sections -shared -Wl,-soname,libLLVMBitWriter.so.20.0git -o lib/libLLVMBitWriter.so.20.0git lib/Bitcode/Writer/CMakeFiles/LLVMBitWriter.dir/BitWriter.cpp.o lib/Bitcode/Writer/CMakeFiles/LLVMBitWriter.dir/BitcodeWriter.cpp.o lib/Bitcode/Writer/CMakeFiles/LLVMBitWriter.dir/BitcodeWriterPass.cpp.o lib/Bitcode/Writer/CMakeFiles/LLVMBitWriter.dir/ValueEnumerator.cpp.o  -Wl,-rpath,"\$ORIGIN/../lib:/vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/lib:"  lib/libLLVMAnalysis.so.20.0git  lib/libLLVMObject.so.20.0git  lib/libLLVMCore.so.20.0git  lib/libLLVMMC.so.20.0git  lib/libLLVMTargetParser.so.20.0git  lib/libLLVMSupport.so.20.0git  -Wl,-rpath-link,/vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/lib && :
ld.lld: error: undefined symbol: llvm::DenseMap<unsigned int, llvm::memprof::FrameStat, llvm::DenseMapInfo<unsigned int, void>, llvm::detail::DenseMapPair<unsigned int, llvm::memprof::FrameStat> > llvm::memprof::computeFrameHistogram<unsigned int>(llvm::MapVector<unsigned long, llvm::SmallVector<unsigned int, CalculateSmallVectorDefaultInlinedElements<unsigned int>::value>, llvm::DenseMap<unsigned long, unsigned int, llvm::DenseMapInfo<unsigned long, void>, llvm::detail::DenseMapPair<unsigned long, unsigned int> >, llvm::SmallVector<std::pair<unsigned long, llvm::SmallVector<unsigned int, CalculateSmallVectorDefaultInlinedElements<unsigned int>::value> >, 0u> >&)
>>> referenced by BitcodeWriter.cpp
>>>               lib/Bitcode/Writer/CMakeFiles/LLVMBitWriter.dir/BitcodeWriter.cpp.o:(writeMemoryProfileRadixTree(llvm::MapVector<unsigned long, llvm::SmallVector<unsigned int, 12u>, llvm::DenseMap<unsigned long, unsigned int, llvm::DenseMapInfo<unsigned long, void>, llvm::detail::DenseMapPair<unsigned long, unsigned int> >, llvm::SmallVector<std::pair<unsigned long, llvm::SmallVector<unsigned int, 12u> >, 0u> >&&, llvm::BitstreamWriter&, unsigned int))

ld.lld: error: undefined symbol: llvm::memprof::CallStackRadixTreeBuilder<unsigned int>::build(llvm::MapVector<unsigned long, llvm::SmallVector<unsigned int, 12u>, llvm::DenseMap<unsigned long, unsigned int, llvm::DenseMapInfo<unsigned long, void>, llvm::detail::DenseMapPair<unsigned long, unsigned int> >, llvm::SmallVector<std::pair<unsigned long, llvm::SmallVector<unsigned int, 12u> >, 0u> >&&, std::optional<llvm::DenseMap<unsigned int, unsigned int, llvm::DenseMapInfo<unsigned int, void>, llvm::detail::DenseMapPair<unsigned int, unsigned int> > const>, llvm::DenseMap<unsigned int, llvm::memprof::FrameStat, llvm::DenseMapInfo<unsigned int, void>, llvm::detail::DenseMapPair<unsigned int, llvm::memprof::FrameStat> >&)
>>> referenced by BitcodeWriter.cpp
>>>               lib/Bitcode/Writer/CMakeFiles/LLVMBitWriter.dir/BitcodeWriter.cpp.o:(writeMemoryProfileRadixTree(llvm::MapVector<unsigned long, llvm::SmallVector<unsigned int, 12u>, llvm::DenseMap<unsigned long, unsigned int, llvm::DenseMapInfo<unsigned long, void>, llvm::detail::DenseMapPair<unsigned long, unsigned int> >, llvm::SmallVector<std::pair<unsigned long, llvm::SmallVector<unsigned int, 12u> >, 0u> >&&, llvm::BitstreamWriter&, unsigned int))
clang: error: linker command failed with exit code 1 (use -v to see invocation)
36.814 [409/3/4609] Linking CXX shared library lib/libLLVMSandboxIR.so.20.0git
36.849 [409/2/4610] Linking CXX shared library lib/libLLVMTransformUtils.so.20.0git
36.887 [409/1/4611] Linking CXX shared library lib/libLLVMExecutionEngine.so.20.0git
ninja: build stopped: subcommand failed.

```

</details>

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