Speculative Decoding in vLLM on AMD GPUs
This technical post explores the implementation of speculative decoding in vLLM on AMD GPUs to optimize large language model serving. It compares various drafting methods and provides experimental results using AMD Instinct MI300X and MI355X hardware.
Why it matters
Optimizing LLM inference is critical for reducing latency and costs in large-scale AI deployments, particularly on non-NVIDIA hardware.
Table of Contents TL;DR: Speculative decoding allows vLLM to verify multiple drafted tokens in a single target-model pass. In our experiments, its effect on output-token throughput varied across drafting methods and proposal lengths, and also depended on the model family, draft checkpoint, workload, and acceptance behavior.
Large language models support a wide range of applications, but serving them at scale requires careful optimization. Standard autoregressive decoding is the baseline used by most LLM serving systems: the model generates one token, appends it to the sequence, and then uses the updated sequence to generate the next token. This process is simple and reliable, but the serving loop still advances one committed token at a time because output tokens must be produced in strict left-to-right order.
Get smarter about the news
Sign up free for a feed built around what you actually care about, Dive Deeper research on any story, and the full text of every article.
Create free accountAlready have an account? Sign in