jax.profiler
模块#
追踪和时间性能分析#
性能分析计算 描述了如何使用 JAX 的追踪和时间性能分析功能。
|
在端口 port 上启动 profiler 服务器。 |
|
启动 profiler 追踪。 |
停止当前正在运行的 profiler 追踪。 |
|
|
用于进行 profiler 追踪的上下文管理器。 |
|
为函数执行生成追踪事件的装饰器。 |
在 profiler 中生成追踪事件的上下文管理器。 |
|
|
在 profiler 中生成步骤追踪事件的上下文管理器。 |
设备内存性能分析#
请参阅 性能分析设备内存 以了解 JAX 设备内存性能分析功能的介绍。
|
将 JAX 设备内存性能分析捕获为 |
|
收集设备内存性能分析并将其写入文件。 |