thunder.executors.passes.del_last_used

thunder.executors.passes.del_last_used(trace, *, clear_mutable_collections=False)[source]
Mark last used intermediates to be deleted. This lets the Python garbage collector free

unused tensor memory.

Parameters:
  • trace (TraceCtx) – trace to be transformed

  • clear_mutable_collections – whether to clear collections

  • trace (TraceCtx) –

Returns:

transformed trace

Return type:

list