llvm
diff --git a/‎llvm/include/llvm/Analysis/MemoryProfileInfo.h
Copy file name to clipboardExpand all lines: llvm/include/llvm/Analysis/MemoryProfileInfo.h
+1-2Lines changed: 1 addition & 2 deletions b/‎llvm/include/llvm/Analysis/MemoryProfileInfo.h
Copy file name to clipboardExpand all lines: llvm/include/llvm/Analysis/MemoryProfileInfo.h
+1-2Lines changed: 1 addition & 2 deletions
diff --git a/‎llvm/include/llvm/IR/ModuleSummaryIndex.h
Copy file name to clipboardExpand all lines: llvm/include/llvm/IR/ModuleSummaryIndex.h
+21-1Lines changed: 21 additions & 1 deletion b/‎llvm/include/llvm/IR/ModuleSummaryIndex.h
Copy file name to clipboardExpand all lines: llvm/include/llvm/IR/ModuleSummaryIndex.h
+21-1Lines changed: 21 additions & 1 deletion
diff --git a/‎llvm/include/llvm/ProfileData/MemProfCommon.h
Copy file name to clipboardExpand all lines: llvm/include/llvm/ProfileData/MemProfCommon.h
-43Lines changed: 0 additions & 43 deletions b/‎llvm/include/llvm/ProfileData/MemProfCommon.h
Copy file name to clipboardExpand all lines: llvm/include/llvm/ProfileData/MemProfCommon.h
-43Lines changed: 0 additions & 43 deletions
diff --git a/‎llvm/lib/Transforms/Instrumentation/MemProfiler.cpp
Copy file name to clipboardExpand all lines: llvm/lib/Transforms/Instrumentation/MemProfiler.cpp
-1Lines changed: 0 additions & 1 deletion b/‎llvm/lib/Transforms/Instrumentation/MemProfiler.cpp
Copy file name to clipboardExpand all lines: llvm/lib/Transforms/Instrumentation/MemProfiler.cpp
-1Lines changed: 0 additions & 1 deletion
@@ -13,9 +13,8 @@
 #ifndef LLVM_ANALYSIS_MEMORYPROFILEINFO_H
 #define LLVM_ANALYSIS_MEMORYPROFILEINFO_H
 
-#include "llvm/IR/InstrTypes.h"
 #include "llvm/IR/Metadata.h"
-#include "llvm/ProfileData/MemProfCommon.h"
+#include "llvm/IR/ModuleSummaryIndex.h"
 #include <map>
 
 namespace llvm {
 
@@ -27,7 +27,6 @@
 #include "llvm/IR/ConstantRange.h"
 #include "llvm/IR/GlobalValue.h"
 #include "llvm/IR/Module.h"
-#include "llvm/ProfileData/MemProfCommon.h"
 #include "llvm/Support/Allocator.h"
 #include "llvm/Support/Compiler.h"
 #include "llvm/Support/InterleavedRange.h"
@@ -307,6 +306,14 @@ template <> struct DenseMapInfo<ValueInfo> {
   static unsigned getHashValue(ValueInfo I) { return hash_value(I.getRef()); }
 };
 
+// For optional hinted size reporting, holds a pair of the full stack id
+// (pre-trimming, from the full context in the profile), and the associated
+// total profiled size.
+struct ContextTotalSize {
+  uint64_t FullStackId;
+  uint64_t TotalSize;
+};
+
 /// Summary of memprof callsite metadata.
 struct CallsiteInfo {
   // Actual callee function.
@@ -343,6 +350,19 @@ inline raw_ostream &operator<<(raw_ostream &OS, const CallsiteInfo &SNI) {
   return OS;
 }
 
+// Allocation type assigned to an allocation reached by a given context.
+// More can be added, now this is cold, notcold and hot.
+// Values should be powers of two so that they can be ORed, in particular to
+// track allocations that have different behavior with different calling
+// contexts.
+enum class AllocationType : uint8_t {
+  None = 0,
+  NotCold = 1,
+  Cold = 2,
+  Hot = 4,
+  All = 7 // This should always be set to the OR of all values.
+};
+
 /// Summary of a single MIB in a memprof metadata on allocations.
 struct MIBInfo {
   // The allocation type for this profiled context.
 
@@ -46,7 +46,6 @@
 #include "llvm/Transforms/Utils/ModuleUtils.h"
 #include <map>
 #include <set>
-#include <unordered_set>
 
 using namespace llvm;
 using namespace llvm::memprof;