The research introduces Intra-Prompt Parallel Decoding (IPPD), a method for parallelizing inference in common-context question answering (CCQA) tasks. The approach addresses the limitation of existing techniques that separate questions across independent prompts, leading to underutilized GPU resources. IPPD achieves this by decoding the next token for each question in a single inference step, sharing memory and computation during the attention process. It utilizes virtual position IDs and attention mask manipulation to produce outputs equivalent to standard prompting, without requiring model fine-tuning or architectural changes. The method is compatible with batched inference, even with differing contexts within prompts.
Source: https://arxiv.org/abs/2609.05707