[bug] Add a mutex to protect counters_map in Statistics (#6645)

Issue: fixes #6644 ### Brief Summary
taichi-dev · Nov 18, 2022 · 559097a · 559097a
1 parent bc08773
commit 559097a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/taichi/util/statistics.cpp b/taichi/util/statistics.cpp
@@ -5,6 +5,7 @@ namespace taichi {
 Statistics stat;
 
 void Statistics::add(std::string key, Statistics::value_type value) {
+  std::lock_guard<std::mutex> _(counters_map_mutex_);
   counters_[key] += value;
 }
 

diff --git a/taichi/util/statistics.h b/taichi/util/statistics.h
@@ -1,4 +1,5 @@
 #include <unordered_map>
+#include <mutex>
 
 #include "taichi/common/core.h"
 
@@ -23,6 +24,7 @@ class Statistics {
 
  private:
   counters_map counters_;
+  std::mutex counters_map_mutex_;
 };
 
 extern Statistics stat;