Revamp handling of subcomputations in HLO graph dumper.
Before, we relied on a hacky heuristic -- "recurse into nested fusion nodes" -- that didn't work for the case when e.g. a fusion node was nested inside a while loop. This change also adds a (very basic) testcase for the HLO graph dumper. PiperOrigin-RevId: 169731958
Loading
Please sign in to comment