add block_manager_v2.py into setup_cython: block_manager_v2 is used w…

…hen num-scheduler-steps>1 (#210)
ROCm · Sep 26, 2024 · 5c50fca · 5c50fca
1 parent a5d87a1
commit 5c50fca
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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 += [