Files
vllm/cacheflow/model_executor/layers/sampler.py