thunder.last_traces

thunder.last_traces(fn)[source]

Obtains the list of computation traces that have been produced for the last run of the function. This is a list of traces mirroring the progression of transformations being applied to the trace (at index 0) that has been acquired from interpreting the user program.

If the function has forward and backward, the forward is returned.

Return type:

list[TraceCtx]