vllm/model_executor/layers/batch_invariant.py · ded8ada86a3962477433054debbcef1d45161850 · OpenDAS / vllm_cscc · GitLab

Find file Blame History Permalink

Add more dims for batch invariant shims (#27489) · ded8ada8

Bram Wasti authored Oct 30, 2025

Signed-off-by: Bram Wasti <bwasti@meta.com>
Signed-off-by: Bram Wasti <bwasti@fb.com>
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

ded8ada8

batch_invariant.py 25.6 KB