From 8a4180c8b6c5b11cc48274daa02ac2c03115514e Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Sat, 21 Dec 2024 17:11:00 -0800 Subject: [PATCH] yapf Signed-off-by: Woosuk Kwon --- vllm/v1/worker/gpu_block_table.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vllm/v1/worker/gpu_block_table.py b/vllm/v1/worker/gpu_block_table.py index 7639bac05c6d6..25a927c371566 100644 --- a/vllm/v1/worker/gpu_block_table.py +++ b/vllm/v1/worker/gpu_block_table.py @@ -93,8 +93,8 @@ class BlockTable: # 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)