TetriServe: Efficiently Serving Mixed DiT Workloads
Step-level sequence parallelism and round-based scheduling improve SLO attainment by up to 32% across heterogeneous DiT workloads.
BibTeX
@inproceedings{lu2026tetriserve,
title = {TetriServe: Efficiently Serving Mixed DiT Workloads},
author = {Lu, Runyu and He, Shiqi and Tan, Wenxuan and Li, Shenggui and Wu, Ruofan and Ma, Jeff J. and Chen, Ang and Chowdhury, Mosharaf},
booktitle = {Proceedings of the 31st ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Volume 2},
pages = {1982--1997},
year = {2026},
doi = {10.1145/3779212.3790233}
}
Abstract
Diffusion Transformer (DiT) models excel at generating high-quality images through iterative denoising steps, but serving them under strict Service Level Objectives (SLOs) is challenging due to their high computational cost, particularly at larger resolutions. Existing serving systems use fixed-degree sequence parallelism, which is inefficient for heterogeneous workloads with mixed resolutions and deadlines, leading to poor GPU utilization and low SLO attainment. In this paper, we propose step-level sequence parallelism to dynamically adjust the degree of parallelism of individual requests according to their deadlines. We present TetriServe, a DiT serving system that implements this strategy for highly efficient image generation. Specifically, TetriServe introduces a novel round-based scheduling mechanism that improves SLO attainment by (1) discretizing time into fixed rounds to make deadline-aware scheduling tractable, (2) adapting parallelism at the step level and minimizing GPU hour consumption, and (3) jointly packing requests to minimize late completions. Extensive evaluation on state-of-the-art DiT models shows that TetriServe achieves up to 32% higher SLO attainment compared to existing solutions without degrading image quality.