mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove backward compatibility logic for tool naming.

Browse Source 
PiperOrigin-RevId: 741030788
This commit is contained in:
Mudit Gokhale 2025-03-26 23:12:32 -07:00 committed by jax authors
parent 8f25337a9f
commit 0c1f4c155e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/collect_profile.py
View File

@ -91,7 +91,7 @@ def collect_profile(port: int, duration_in_ms: int, host: str,
in root_trace_folder.iterdir()]
latest_folder = max(trace_folders, key=os.path.getmtime)
xplane = next(latest_folder.glob("*.xplane.pb"))
result, _ = convert.xspace_to_tool_data([xplane], "trace_viewer^", {})
result, _ = convert.xspace_to_tool_data([xplane], "trace_viewer", {})
with gzip.open(str(latest_folder / "remote.trace.json.gz"), "wb") as fp:
fp.write(result.encode("utf-8"))