[Bugfix] Fix CPU backend crash in KV cache block zeroing (#37550)

Signed-off-by: DorBernsohn <dor.bernsohn@gmail.com>
This commit is contained in:
DorBernsohn
2026-03-23 13:35:45 +02:00
committed by GitHub
parent debd6e768c
commit 7938d12119

View File

@@ -88,6 +88,11 @@ class CPUModelRunner(GPUModelRunner):
def _sync_device(self) -> None:
pass
def _zero_block_ids(self, block_ids: list[int]) -> None:
# CPU attention assigns -INF to logits at invalid positions,
# so stale KV cache data never affects computation.
pass
def get_dp_padding(self, num_tokens: int) -> tuple[int, torch.Tensor | None]:
# Note: For CPU backend, dp padding is not required for now.
return 0, None