Files
vllm/vllm/model_executor/layers/batch_invariant.py