More debugging

wujingyue · wujingyue · commit 49552019e428 · 2026-01-28T17:01:14.000-08:00
diff --git a/csrc/preseg_passes/propagate_shardings.cpp b/csrc/preseg_passes/propagate_shardings.cpp
@@ -187,6 +187,13 @@ void PropagateShardingsPass::runPass(Fusion* fusion) {
           PropagateDirection::kBackward);
     }
   }
+
+  if (isDebugDumpEnabled(DebugDumpOption::PreSegmenterLogging)) {
+    debug() << std::endl
+            << "Fusion Transforms after " << name() << ":" << std::endl;
+    fusion->printTransforms();
+    debug() << std::endl;
+  }
 }
 
 } // namespace nvfuser::preseg_passes

Original file line number	Diff line number	Diff line change
`@@ -187,6 +187,13 @@ void PropagateShardingsPass::runPass(Fusion* fusion) {`
`187`	`187`	`PropagateDirection::kBackward);`
`188`	`188`	`}`
`189`	`189`	`}`
	`190`	`+`
	`191`	`+ if (isDebugDumpEnabled(DebugDumpOption::PreSegmenterLogging)) {`
	`192`	`+ debug() << std::endl`
	`193`	`+ << "Fusion Transforms after " << name() << ":" << std::endl;`
	`194`	`+ fusion->printTransforms();`
	`195`	`+ debug() << std::endl;`
	`196`	`+ }`
`190`	`197`	`}`
`191`	`198`
`192`	`199`	`} // namespace nvfuser::preseg_passes`