Hybrid JIT-CUDA Graph Optimization for Low-Latency Large Language Model Inference

View at publisher
Authors: Divakar Kumar Yadav, Tian Zhao
Cited by: 0

Abstract

Large Language Models (LLMs) deliver strong performance but suffer from high inference latency and unstable kernel-launch overhead. This the- sis introduces a Hybrid JIT–CUDA Graph Runtime that combines static CUDA Graph replay with JIT-compiled dynamic kernels to achieve deter- ministic, low-latency autoregressive decoding. A rolling graph-generation mechanism enables variable sequence lengths while preserving static-shape constraints. Evaluated on an NVIDIA H100 GPU, the system reduces Time- to-First-Token (TTFT) by 45–80%, achieves the lowest and most stable P99 latency, and provides the fastest end-to-end 500-token generation compared with HuggingFace (PyTorch Eager) and TensorRT–LLM. The results demon- strate that integrating JIT flexibility with CUDA Graph determinism offers an effective approach for high-performance LLM inference.

Fields of Study & Subjects

Parallel Computing and Optimization Techniques Natural Language Processing Techniques Topic Modeling

Publication details

Journal / source
IEEE Annual International Computer Software and Applications Conference (COMPSAC)
Pages
196–204
Publication date
1 July 2026
Language
EN