From 5c50fcad362c414ee27c876d9ea022089238b9b5 Mon Sep 17 00:00:00 2001 From: sanyalington Date: Fri, 27 Sep 2024 02:24:38 +0530 Subject: [PATCH] add block_manager_v2.py into setup_cython: block_manager_v2 is used when num-scheduler-steps>1 (#210) --- setup_cython.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup_cython.py b/setup_cython.py index dca79af61a9f6..30848fa042d8b 100644 --- a/setup_cython.py +++ b/setup_cython.py @@ -18,6 +18,7 @@ "vllm/core/scheduler.py", "vllm/sequence.py", "vllm/core/block_manager_v1.py", + "vllm/core/block_manager_v2.py", ] infiles += [