BroadcastOp::toString() show broadcast flags as well (#3641)

For disambiguation --------- Co-authored-by: Jacob Hinkle <[email protected]>
NVIDIA · Dec 24, 2024 · c7f40b7 · c7f40b7
1 parent e214d37
commit c7f40b7
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/csrc/ir/nodes.cpp b/csrc/ir/nodes.cpp
@@ -1268,7 +1268,17 @@ BroadcastOp::BroadcastOp(
 std::string BroadcastOp::toString(int indent_size) const {
   std::stringstream ss;
   indent(ss, indent_size) << out()->toString() << "\n";
-  indent(ss, indent_size) << "   = broadcast( " << in()->toString() << " )\n";
+  indent(ss, indent_size) << "   = broadcast( " << in()->toString()
+                          << ", flags = {";
+  bool is_first = true;
+  for (const auto f : getBroadcastDimFlags()) {
+    if (!is_first) {
+      ss << ", ";
+    }
+    ss << (f ? "true" : "false");
+    is_first = false;
+  }
+  ss << "} )\n";
   return ss.str();
 }