SamplePGO - Clear coverage tracking when clearing per-function data.

[oota-llvm.git] / lib / Transforms / IPO / SampleProfile.cpp

diff --git a/lib/Transforms/IPO/SampleProfile.cpp b/lib/Transforms/IPO/SampleProfile.cpp
index 20bfac954a55dcee5ff1a5246805bb8115fed3d1..de4170692fe0a098b775da6e4f21e969abd90972 100644 (file)
--- a/lib/Transforms/IPO/SampleProfile.cpp
+++ b/lib/Transforms/IPO/SampleProfile.cpp
@@ -188,6 +188,7 @@ public:
   unsigned computeCoverage(unsigned Used, unsigned Total) const;
   unsigned countUsedSamples(const FunctionSamples *Samples) const;
   unsigned countBodySamples(const FunctionSamples *Samples) const;
+  void clear() { SampleCoverage.clear(); }
 
 private:
   typedef DenseMap<LineLocation, unsigned> BodySampleCoverageMap;
@@ -285,6 +286,7 @@ void SampleProfileLoader::clearFunctionData() {
   LI = nullptr;
   Predecessors.clear();
   Successors.clear();
+  CoverageTracker.clear();
 }
 
 /// \brief Returns the offset of lineno \p L to head_lineno \p H