In DiT inference for video and image generation, compute-heavy core operators often already have dedicated optimized kernels, yet many small operator chains still lack mature, efficient optimizations—memory traffic and kernel launch overhead remain non-trivial. Hand-written Triton can optimize hot spots in those chains, but each pattern requires separate development and maintenance; using `torch.compile` directly often yields unstable gains or even regressions when subgraph boundaries and dynamic shapes are not handled carefully.
0 commit comments