[feature] Add start step profile argument in /start_profile (#7608)

This commit is contained in:
kyleliang-nv
2025-07-09 18:42:15 -07:00
committed by GitHub
parent 7590f5224b
commit dd445a41f5
6 changed files with 142 additions and 1 deletions

View File

@@ -418,6 +418,7 @@ async def start_profile_async(obj: Optional[ProfileReqInput] = None):
await _global_state.tokenizer_manager.start_profile(
output_dir=obj.output_dir,
start_step=obj.start_step,
num_steps=obj.num_steps,
activities=obj.activities,
with_stack=obj.with_stack,