Revert "removing mtx"

This reverts commit ef2047b.
triton-inference-server · Jul 2, 2024 · 8bef853 · 8bef853
1 parent a89aa19
commit 8bef853
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/infer_trace.cc b/src/infer_trace.cc
@@ -59,6 +59,7 @@ void
 InferenceTrace::RecordActivityName(
     uint64_t timestamp_ns, std::string activity_name)
 {
+  std::lock_guard<std::mutex> lock(mu_);
   triton::common::TritonJson::Value context_json(
       triton::common::TritonJson::ValueType::OBJECT);
   if (!context_.empty()) {

diff --git a/src/infer_trace.h b/src/infer_trace.h
@@ -137,6 +137,7 @@ class InferenceTrace {
   // across traces
   static std::atomic<uint64_t> next_id_;
   std::string context_ = "";
+  std::mutex mu_;
 };
 
 //