aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--include/llvm/ADT/StringMap.h6
-rw-r--r--lib/CodeGen/CollectorMetadata.cpp5
2 files changed, 7 insertions, 4 deletions
diff --git a/include/llvm/ADT/StringMap.h b/include/llvm/ADT/StringMap.h
index f180b36360..7189931abb 100644
--- a/include/llvm/ADT/StringMap.h
+++ b/include/llvm/ADT/StringMap.h
@@ -333,6 +333,12 @@ public:
return true;
}
+ // clear - Empties out the StringMap
+ void clear() {
+ while (!empty())
+ erase(begin());
+ }
+
/// GetOrCreateValue - Look up the specified key in the table. If a value
/// exists, return it. Otherwise, default construct a value, insert it, and
/// return.
diff --git a/lib/CodeGen/CollectorMetadata.cpp b/lib/CodeGen/CollectorMetadata.cpp
index 7a5a699a11..5b8c7c7999 100644
--- a/lib/CodeGen/CollectorMetadata.cpp
+++ b/lib/CodeGen/CollectorMetadata.cpp
@@ -112,10 +112,7 @@ CollectorMetadata &CollectorModuleMetadata::get(const Function &F) {
void CollectorModuleMetadata::clear() {
Map.clear();
-
- // TODO: StringMap should provide a clear method.
- while (!NameMap.empty())
- NameMap.erase(NameMap.begin());
+ NameMap.clear();
for (iterator I = begin(), E = end(); I != E; ++I)
delete *I;