benchmarks.SaturnNPU.scripts.plot_sankey
Source: benchmarks/SaturnNPU/scripts/plot_sankey.py
benchmarks.SaturnNPU.scripts.plot_sankey
Generate a Sankey diagram showing how PyTorch modules decompose through MLIR.
Flow: PyTorch modules → sub-modules → Torch-MLIR ops → Linalg/Input → Global-Opt
Usage
python scripts/plot_sankey.py --output-dir benchmarks/SaturnNPU/plots/