feat(//core/partitioning): Add an ostream implementation for

PartitionInfo

Signed-off-by: Naren Dasan <naren@narendasan.com>
Signed-off-by: Naren Dasan <narens@nvidia.com>

Author: narendasan

core/partitioning/BUILD

@@ -19,6 +19,7 @@ cc_library(
         "SegmentedBlock.cpp",
         "shape_analysis.cpp",
         "partitioning.cpp",
+        "PartitionInfo.cpp",
     ],
     deps = [
         "//core/util:prelude",

core/partitioning/PartitionInfo.cpp

@@ -0,0 +1,30 @@
+#include <iostream>
+#include <sstream>
+#include <utility>
+
+#include "core/partitioning/PartitionInfo.h"
+
+namespace trtorch {
+namespace core {
+namespace partitioning {
+// clang-format off
+std::ostream& operator<<(std::ostream& os, const PartitionInfo& s) {
+  os << "Settings requested for Torch Fallback:" \
+     << "\n    \"enabled\": ";
+  if (s.enabled) {
+    os << "True";
+    os << "\n    \"min_block_size\": " << s.min_block_size \
+       << "\n    \"forced_fallback_operators\": [";
+    for (auto i : s.forced_fallback_operators) {
+      os <<"\n        " << i << ',';
+    }
+    os << "\n     ]";
+  } else {
+    os << "False";
+  }
+  return os;
+}
+// clang-format on
+} // namespace partitioning
+} // namespace core
+} // namespace trtorch