Skip to main content

Module graph_io

Module graph_io 

Source
Expand description

Static IO / sync profile for compiled graphs (Phase 0 — fusion planning).

Structs§

GraphIoOptions
Tuning for static IO analysis.
GraphIoProfile
Host-visible traffic and sync points for one forward pass.

Functions§

metal_host_sync_kinds
Ops that force a GPU sync + host thunk on Metal today.
profile_graph_io
Profile a graph before compile (conservative static estimate).
profile_graph_io_outputs
Profile with only selected outputs materialized on the host (peaks-only, logits-only, …).
profile_graph_io_with_options