CollTrace Scuba Logging for Dashboard (#8)

Summary:

D82675147 got reverted because of conflict with NCCLX OSS. In this new version, we fix this by not importing RCCLX code into the common folder.

Differential Revision: D85305175

Author: ahmd-k

comms/utils/logger/ScubaLogger.cc

@@ -67,6 +67,13 @@ void NcclScubaEvent::record() {
   ncclLogToScuba(type_, copySample);
 }
 
+void NcclScubaEvent::record(const std::string& stage) {
+  if (!stage.empty()) {
+    sample_.addNormal("stage", stage);
+  }
+  record();
+}
+
 void NcclScubaEvent::setLogMetatData(const CommLogData* logMetaData) {
   sample_.setCommunicatorMetadata(logMetaData);
 }

comms/utils/logger/ScubaLogger.h

@@ -12,12 +12,12 @@
 #include "comms/utils/logger/EventMgr.h"
 #include "comms/utils/logger/NcclScubaSample.h"
 
-class NcclScubaEvent {
- public:
+struct NcclScubaEvent {
   void startAndRecord();
   void stopAndRecord();
   void lapAndRecord(const std::string& stage = "");
   void record();
+  void record(const std::string& stage);
   void setLogMetatData(const CommLogData* logMetaData);
 
   explicit NcclScubaEvent(const std::string& stage);
@@ -26,7 +26,6 @@ class NcclScubaEvent {
 
   NcclScubaEvent(const std::string& stage, const CommLogData* logMetaData);
 
- private:
   NcclScubaSample sample_;
   folly::stop_watch<std::chrono::microseconds> timer_;
   std::string stage_{};