add JSON-formatted metadata for ClientTraceActivity

libkineto/CMakeLists.txt

@@ -112,6 +112,7 @@ target_include_directories(kineto_base PUBLIC
       $<BUILD_INTERFACE:${CUDA_INCLUDE_DIRS}>)
 
 target_include_directories(kineto_api PUBLIC
+      $<BUILD_INTERFACE:${FMT_INCLUDE_DIR}>
       $<BUILD_INTERFACE:${LIBKINETO_INCLUDE_DIR}>)
 
 if(KINETO_LIBRARY_TYPE STREQUAL "default")

libkineto/include/ClientTraceActivity.h

@@ -7,8 +7,10 @@
 
 #pragma once
 
+#include <fmt/format.h>
 #include <string>
 #include <thread>
+#include <vector>
 
 #include "TraceActivity.h"
 
@@ -56,6 +58,16 @@ struct ClientTraceActivity : TraceActivity {
     // Unimplemented by default
   }
 
+  // Encode client side metadata as a key/value string.
+  void addMetadata(const std::string& key, const std::string& value) {
+    auto kv = fmt::format("\"{}\": {}", key, value);
+    metadata_.push_back(std::move(kv));
+  }
+
+  [[nodiscard]] const std::string getMetadata() const {
+    return fmt::format("{}", fmt::join(metadata_, ", "));
+  }
+
   int64_t startTime{0};
   int64_t endTime{0};
   int64_t correlation{0};
@@ -72,6 +84,9 @@ struct ClientTraceActivity : TraceActivity {
   std::string inputNames;
   std::string outputNames;
   std::string callStack;
+
+ private:
+  std::vector<std::string> metadata_;
 };
 
 } // namespace libkineto

libkineto/libkineto_defs.bzl

@@ -53,6 +53,7 @@ def get_libkineto_public_headers():
     return [
         "include/ActivityProfilerInterface.h",
         "include/ActivityType.h",
+        "include/ClientTraceActivity.h",
         "include/ClientInterface.h",
         "include/TraceActivity.h",
         "include/TraceSpan.h",