thunder.executors.passes.update_fusion_call_ctx

thunder.executors.passes.update_fusion_call_ctx(trace)

Updates the call context of the trace to be the current call context.

Some of the fusion bound symbols may have been created with a different call context and modified after the fact. This pass ensures that the call context is correct.

Parameters:

• trace (TraceCtx) – trace to be transformed

• trace (TraceCtx) –

Returns:

transformed trace

Return type:

(TraceCtx)