reduce usage of steady_vector::size_

Oneflow-Inc · Jun 14, 2022 · b2a808b · b2a808b
1 parent e7ae284
commit b2a808b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/oneflow/core/common/steady_vector.h b/oneflow/core/common/steady_vector.h
@@ -53,12 +53,13 @@ class SteadyVector {
 
   void push_back(const T& elem) {
     std::unique_lock<std::mutex> lock(mutex_);
-    int granularity = GetGranularity(size_);
-    if (size_ + 1 == (1 << granularity)) {
+    size_t size = size_;
+    int granularity = GetGranularity(size);
+    if (size + 1 == (1 << granularity)) {
       CHECK_LT(granularity, N);
       granularity2data_[granularity].reset(new T[1 << granularity]);
     }
-    this->Mut(size_) = elem;
+    this->Mut(size) = elem;
     ++size_;
   }