Toggle navigation

.rst .pdf

Warning

This page was created from a pull request (#9655).

jax.profiler.stop_trace

jax.profiler.stop_trace()

Stops the currently-running profiler trace.

The trace will be saved to the log_dir passed to the corresponding start_trace() call. Raises a RuntimeError if a trace hasn’t been started.

previous

jax.profiler.start_trace

next

jax.profiler.trace