Connect collector to reporter

triton-inference-server · Aug 2, 2023 · 14ffbf2 · 14ffbf2
1 parent 9cfcbd2
commit 14ffbf2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/c++/perf_analyzer/perf_analyzer.cc b/src/c++/perf_analyzer/perf_analyzer.cc
@@ -434,6 +434,7 @@ PerfAnalyzer::WriteReport()
 void
 PerfAnalyzer::GenerateProfileExportReport()
 {
+  reporter_->ConvertToJson(collector_->GetData(), collector_->GetVersion());
   reporter_->OutputToFile(params_->profile_export_file);
 }
 

diff --git a/src/c++/perf_analyzer/raw_data_collector.h b/src/c++/perf_analyzer/raw_data_collector.h
@@ -83,6 +83,12 @@ class RawDataCollector {
   /// @param request_records The request information for the current experiment.
   void AddData(PerfMode& id, std::vector<RequestRecord>&& request_records);
 
+  /// Get the raw experiment data for the profile
+  /// @return Raw experiment data
+  std::vector<Experiment>& GetData() { return experiments_; }
+
+  std::string& GetVersion() { return version_; }
+
  private:
   RawDataCollector() = default;
 
@@ -94,6 +100,6 @@ class RawDataCollector {
   };
 
   std::vector<Experiment> experiments_{};
-  std::string version{VERSION};
+  std::string version_{VERSION};
 };
 }}  // namespace triton::perfanalyzer