Skip to content
Trending storyDeveloping

vLLM reuses its inference kernels in the training forward pass for matching results

1 article1 sourcesince Oct 15Last article 1h ago ·

Overview

AISummary of 1 article

vLLM says its training forward pass now calls its inference kernels directly.

The stated aim is that the same token gets the same result across validated batch sizes, phases and parallel layouts. The kernels aligned are FP8 GEMM, sparse-attention Top-K, FP8 KV-cache reads, MoE communication and CUDA Graph execution.

The backward pass does not share these kernels. vLLM says it uses separate checks for the backward pass.

Written by AI from the articles below · updated Oct 15, 10:57 AM ET

Check the sources:

Article timeline

The articles in this story. Times are ET.

Oct 15
  1. vLLMOfficial
    vLLM training forward pass reuses its inference kernels for matching results

    AIvLLM says its training forward pass calls its inference kernels directly, aligning FP8 GEMM, sparse-attention Top-K, FP8 KV-cache reads, MoE communication and CUDA Graph execution. The aim is for the same token to get the same result across validated batch sizes, phases and parallel layouts. The backward pass uses separate checks.

    Image from @vllm_project's post

Heat trend

Not enough continuous observations to show a trend yet.