jax.profiler.stop_trace

jax.profiler.stop_trace()

停止当前正在运行的性能分析器跟踪。

跟踪结果将被保存到传递给相应 start_trace() 调用的 log_dir 中。 如果没有开始跟踪，则会引发 RuntimeError。