[Doc]: fix typos in Python comments (#24093)

Signed-off-by: Didier Durand <durand.didier@gmail.com>
This commit is contained in:
Didier Durand
2025-09-03 06:05:45 +02:00
committed by GitHub
parent c4ed78b14f
commit d7e1e59972
15 changed files with 23 additions and 23 deletions

View File

@@ -187,7 +187,7 @@ def test_tree_attn_correctness() -> None:
dtype=torch.bfloat16,
)
# Setup the block table and KV cache for paged KV.
# Set up the block table and KV cache for paged KV.
assert max_sequence_length % block_size == 0
max_blocks_per_batch = max_sequence_length // block_size
kv_cache = torch.randn(
@@ -222,7 +222,7 @@ def test_tree_attn_correctness() -> None:
num_alloc_blocks_per_batch] = block_ids.view(
-1, num_alloc_blocks_per_batch)
# Setup the slot mapping for the input KVs.
# Set up the slot mapping for the input KVs.
tree_positions = sequence_position + torch.arange(
0,
tree_size_q,