thunder.core.transforms.insert_inplace

thunder.core.transforms.insert_inplace(trc, idx, fn)[source]

Calls fn and record any symbols called into trc, starting at idx.

Parameters:

trc (TraceCtx) – Trace to insert BoundSymbols representing fn.
idx (int) – Starting index of trc.bound_symbols to insert BoundSymbols representing fn.
fn (Callable[[], Any]) –
trc (TraceCtx) –
idx (int) –
fn (Callable[[], Any]) –

Return type:

None

Note

This operation is inplace. It will modify the given trc’s bound_symbols.

Note

Because this operation is explicitly inplace, it will disregard whether or not mark_complete() has been called on trc already.

Was this page helpful?