thunder.core.transforms.insert_inplace

thunder.core.transforms.insert_inplace(trc, idx, fn)

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.