fix rebase to 80dd8fe

johncalesp · johncalesp · commit 75110106ba94 · 2025-09-18T18:12:25.000Z
Signed-off-by: John Calderon &lt;jcalderon@nvidia.com&gt;
diff --git a/tensorrt_llm/_torch/pyexecutor/py_executor_creator.py b/tensorrt_llm/_torch/pyexecutor/py_executor_creator.py
@@ -575,6 +575,7 @@ def drafting_loop_wrapper(model):
             kv_cache_config=kv_cache_config,
             pytorch_backend_config=pytorch_backend_config,
             speculative_config=spec_config,
+            profiling_stage_data=profiling_stage_data,
         )
         estimating_kv_cache = kv_cache_creator.try_prepare_estimation()
         with mem_monitor.observe_creation_stage(

Original file line number	Diff line number	Diff line change
`@@ -575,6 +575,7 @@ def drafting_loop_wrapper(model):`
`575`	`575`	`kv_cache_config=kv_cache_config,`
`576`	`576`	`pytorch_backend_config=pytorch_backend_config,`
`577`	`577`	`speculative_config=spec_config,`
	`578`	`+ profiling_stage_data=profiling_stage_data,`
`578`	`579`	`)`
`579`	`580`	`estimating_kv_cache = kv_cache_creator.try_prepare_estimation()`
`580`	`581`	`with mem_monitor.observe_creation_stage(`