Skip to content

Commit fdfacbc

Browse files
committed
fix
Signed-off-by: junq <[email protected]>
1 parent 57b9db1 commit fdfacbc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorrt_llm/_torch/pyexecutor/model_engine.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2027,7 +2027,7 @@ def forward(
20272027
self.iter_counter += 1
20282028

20292029
# 3. Ask the manager to execute the graph
2030-
graph_output = self.cuda_graph_manager.execute(
2030+
graph_output = self.cuda_graph_model_engine.execute(
20312031
batch=padded_requests,
20322032
inputs=inputs,
20332033
forward_fn=self._forward_step)

0 commit comments

Comments
 (0)