yapf

Signed-off-by: Woosuk Kwon <[email protected]>
vllm-project · Dec 22, 2024 · 8a4180c · 8a4180c
1 parent 1aaced5
commit 8a4180c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vllm/v1/worker/gpu_block_table.py b/vllm/v1/worker/gpu_block_table.py
@@ -93,8 +93,8 @@ def apply_diff(self, num_reqs: int) -> None:
             # Copy the entire block table to the GPU.
             # NOTE(woosuk): This can be a performance bottleneck when the block
             # table is large.
-            self.block_table[:num_reqs].copy_(
-                self.block_table_cpu[:num_reqs], non_blocking=True)
+            self.block_table[:num_reqs].copy_(self.block_table_cpu[:num_reqs],
+                                              non_blocking=True)
 
     def clear(self) -> None:
         self.block_table.fill_(0)