Work

Deep Kernel Fusion for Transformers

Year: 2026
Type: conference-paper
Abstract: Agentic LLM inference with long contexts is increasingly limited by memory bandwidth rather than compute.In this setting, SwiGLU MLP blocks, whose large weights exceed cache capacity, become a major yet under-optimized bottleneck in the Transformer architecture.We propose DeepFusionKernel, a deeply fused kernel that cuts HBM traffic and boosts cache reuse, delivering up to 13.2% speedup on H100 and 9.7% on A100 over SGLang.Integrated with SGLang and paired with a kernel scheduler, DeepFusionKernel ensures consistent accelerations across generation lengths, while remaining adaptable to diverse models, inference configurations, and hardware platforms.1
Language: English
Open Access status: hybrid
OpenAlex ID: W7166884542
FWCI:
Cites:
Cited by:
Related to: 10