karylab_agents/vllm
1
0
Fork 0
forked from Karylab-cklius/vllm
Code Pull Requests 2 Actions 1 Packages Activity

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

Browse Source 
Signed-off-by: Woosuk Kwon <woosuk@inferact.ai>
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: