Add XPU profiler activity support in benchmark code (#12981)

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Kalyan Kumar
2026-03-05 12:52:56 +05:30
committed by GitHub
parent 2bdd89a6cd
commit c1df359b44
5 changed files with 98 additions and 25 deletions

View File

@@ -26,6 +26,7 @@ def run_profile(
profile_by_stage: bool = False,
merge_profiles: bool = False,
profile_prefix: Optional[str] = None,
start_step: Optional[int] = None,
) -> str:
if output_dir is None:
output_dir = PROFILER_DIR
@@ -57,6 +58,8 @@ def run_profile(
"merge_profiles": merge_profiles,
"profile_prefix": profile_prefix,
}
if start_step is not None:
json_data["start_step"] = str(start_step)
response = requests.post(url=url + "/start_profile", json=json_data)
response.raise_for_status()