Optimization for map_data_.clear()

5 years ago · 7c1b44731e
parent 8300802b95
commit 7c1b44731e
2 changed files with 6 additions and 2 deletions
--- a/mindspore/ccsrc/utils/ordered_map.h
+++ b/mindspore/ccsrc/utils/ordered_map.h
@ -88,7 +88,9 @@ class OrderedMap {
  }

  void clear() {
+    if (!map_data_.empty()) {
      map_data_.clear();
+    }
    sequential_data_.clear();
  }

--- a/mindspore/ccsrc/utils/ordered_set.h
+++ b/mindspore/ccsrc/utils/ordered_set.h
@ -127,7 +127,9 @@ class OrderedSet {

  // Clear the elements
  void clear() {
+    if (!mapped_data_.empty()) {
      mapped_data_.clear();
+    }
    ordered_data_.clear();
  }