Skip to content

Commit 1bded43

Browse files
authored
NVFUSER_DUMP=segmented_fusion prints transforms (#5912)
for multi-GPU debugging. Multi-GPU scheduling happens before segmentation and the shardings are encoded as loop transforms.
1 parent fc60235 commit 1bded43

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

csrc/fusion_segmenter.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1732,9 +1732,10 @@ std::ostream& operator<<(
17321732
}
17331733

17341734
void SegmentedFusion::print() const {
1735-
debug() << "Segmented_Fusion Dump: -- Re-written complete fusion:{\n";
1736-
completeFusion()->printMath();
1737-
debug() << "} // {Re-written complete fusion}\n";
1735+
debug() << "Segmented_Fusion Dump: -- Re-written complete fusion:{"
1736+
<< std::endl;
1737+
completeFusion()->print();
1738+
debug() << "} // {Re-written complete fusion}" << std::endl << std::endl;
17381739
debug() << this << "\n";
17391740
}
17401741

0 commit comments

Comments
 (0)