minor: zero workspace buffer init for flashinfer trtllm-gen attn (#22603)

2025-08-15 17:38:10 -04:00
parent 00d6cba0cf
commit 1723ef1aae
3 changed files with 4 additions and 4 deletions
--- a/vllm/v1/attention/backends/flashinfer.py
+++ b/vllm/v1/attention/backends/flashinfer.py
@@ -252,7 +252,7 @@ class FlashInferMetadataBuilder(AttentionMetadataBuilder[FlashInferMetadata]):

    def _get_workspace_buffer(self):
        if self._workspace_buffer is None:
-            self._workspace_buffer = torch.empty(
+            self._workspace_buffer = torch.zeros(
                FLASHINFER_WORKSPACE_BUFFER_SIZE,
                dtype=torch.uint8,
                device=self.device)