---
id: pr-vllm-14227
repo: vllm-project/vllm
pr: 14227
title: '[V1][TPU] Support V1 Sampler for ragged attention'
author: NickLucche
date: '2025-03-20'
url: https://github.com/vllm-project/vllm/pull/14227
source_category: upstream-code
architectures:
- sm100
tags:
- attention
techniques: []
hardware_features: []
kernel_types:
- attention
languages:
- python
captured_at: '2026-04-27'
status: merged
merge_sha: d8c6d7d6b503000e46186d362626135887322006
inclusion_reason: semantic match with file changes
changed_paths:
- tests/v1/tpu/test_sampler.py
- vllm/v1/sample/ops/topk_topp_sampler.py
- vllm/v1/sample/tpu/__init__.py
- vllm/v1/sample/tpu/metadata.py
- vllm/v1/sample/tpu/sampler.py
- vllm/v1/worker/tpu_model_runner.py
head_ref_oid: 4d6d30c7aa0238b23cffd4950ac178ac4356743e
artifact_dir: evidence/pull-bundles/vllm/gh-14227
---
# [V1][TPU] Support V1 Sampler for ragged attention

- Repository: `vllm-project/vllm`
- PR: [14227](https://github.com/vllm-project/vllm/pull/14227)
- Decision: `semantic match with file changes`
- Reason: semantic match with file changes

## Changed Paths

- `tests/v1/tpu/test_sampler.py`
- `vllm/v1/sample/ops/topk_topp_sampler.py`
- `vllm/v1/sample/tpu/__init__.py`
- `vllm/v1/sample/tpu/metadata.py`
- `vllm/v1/sample/tpu/sampler.py`
- `vllm/v1/worker/tpu_model_runner.py`

## Evidence Bundle

`evidence/pull-bundles/vllm/gh-14227`
