jax-ml/jax

Support profiling without source code changes

开放

#20,293 创建于 2024年3月18日

在 GitHub 查看贡献者指南
 (16 条评论) (2 个反应) (0 位负责人)Python (3,741 个派生)github user discovery
enhancementgood first issue

仓库指标

星标
 (36,195 个星标)
PR 合并指标
 (平均合并 2天 19小时) (30 天内合并 439 个 PR)

描述

Currently adding profiling requires source-code changes which could be difficult in a large codebase.

Would be great if something like JAX_PROFILE=output.pb was supported to generate profiling from command line.

CC @hawkinsp

贡献者指南

研究方向
检查当前性能分析是如何在代码库中启动的，并实现一个环境变量，以便在不修改源代码的情况下启用性能分析。重点关注JAX开始执行的入口点。
技术栈
python
领域
performancedeveloper experience
议题类型
功能
难度面向新贡献者的预计实现难度，1 表示很小改动，5 表示专家级工作。
2
预计时间有经验贡献者完成调查、实现、测试并准备 pull request 的粗略时间范围。
1-2 天
活动状态议题当前的可参与程度：新鲜、活跃、陈旧、阻塞或等待维护者输入。
活跃
清晰度议题是否清楚说明期望改动、验收标准和下一步。
清晰
前置要求
Python
新手友好度1-100 的估计分数，表示该议题对首次贡献者的友好程度。
65