thunder.core.transforms.replace_inplace

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

Removes idx-th BoundSymbol of trc and replace it bsyms representing fn.

Parameters:

• trc (TraceCtx) – Trace to insert BoundSymbols representing fn.

• idx (int) – Index of BoundSymbol of trc.

• fn (Callable[[BoundSymbol], Any]) – Callable to bake into trc, instead of idx-th BoundSymbol.

• trc (TraceCtx) –

• idx (int) –

• fn (Callable[[BoundSymbol], 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.