obscura/vllm
2
0
Fork 0
mirror of https://github.com/vllm-project/vllm.git synced 2026-08-18 19:50:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

[MRV2] Remove assignment of graph_pool in cudagraph_utils (#44338)

Browse Source 
Signed-off-by: Woosuk Kwon <[email protected]>
This commit is contained in:
Woosuk Kwon
2026-06-02 11:50:27 -07:00
committed by GitHub
parent b8b49e2395
commit e4a2e584e5
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
vllm/v1/worker/gpu/cudagraph_utils.py
-1
View File
@@ -299,7 +299,6 @@ class CudaGraphManager:
self.breakable_cg_runner = BreakableCUDAGraphWrapper(
model, self.vllm_config
)
self.breakable_cg_runner.graph_pool = self.pool
def run_pw_graph(self, model: nn.Module, model_inputs: dict[str, Any]) -> Any:
if not self.use_breakable_cg: