Yuxuan Jia

SAFE-DiT: Semantics-Aware Fast-path Execution for High-Resolution Diffusion Transformers

Xuanhua Yin, Yuxuan Jia, Chuanzhi Xu, Weidong Cai

High-resolution Diffusion Transformer (DiT) inference contains substantial spatial redundancy, but many spatially adaptive implementations encode regional computation as attention masks, which can inadvertently move scaled dot-product attention (SDPA) away from FlashAttention fas…Read more