Skip to content

Conversation

Puiching-Memory
Copy link
Contributor

@Puiching-Memory Puiching-Memory commented Mar 6, 2025

Description

在这个合并请求中,分支Profiler相比目前的分支添加了:

  • torch.profiler.profile可用的回调函数,每次调用将调用profiler.export_chrome_trace()保存一份json性能追踪报告,将保存至get_run_dir()/files/trace.json下

后续需要云端的解析,或翻译成现有的API进行上传

Closes: #858

@ShaohonChen
Copy link
Contributor

感谢老哥的贡献!我们将继续跟进!

@SAKURA-CAT
Copy link
Member

#1047 合并后,我们将允许保存在本地

@SAKURA-CAT
Copy link
Member

暂时性的,我保存了此 PR 中的 trace 函数,在未来上线此功能时会使用到它!

@SAKURA-CAT SAKURA-CAT merged commit 605284a into SwanHubX:main Jun 8, 2025
5 checks passed
@Puiching-Memory Puiching-Memory deleted the Profiler branch June 8, 2025 15:50
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
💪 enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[REQUEST] 添加torch.profiler支持
3 participants