Efficient multi-tenant LoRA serving via SGMV-specific operator autotuning
摘要
Low-Rank Adaptation (LoRA) has become a mainstream technique for fine-tuning large language models (LLMs) on downstream tasks due to its memory efficiency. However, traditional inference architectures suffer from computational redundancy and low GPU utilization when serving multi-tenant LoRA models. To address these inefficiencies, this paper presents GD2O, an SGMV-specific operator autotuning framework for multi-tenant LoRA serving. GD2O uses a graph representation to organize the construction space of the non-standard Segmented Gather Matrix-Vector Multiplication (SGMV) operator and applies a hardware-aware Markov-style stochastic search to select tiling, caching, and virtual-thread scheduling configurations. The search is used as a practical heuristic for efficient exploration, not as a formal guarantee of global optimality. The text generation experiments on Llama-2 and two GPU architectures (NVIDIA A100 and RTX 4090) demonstrate that GD2O achieves a throughput improvement of 1.29–1.52