Apply set_execution_file
to specific traces such as forward/backward execution traces
#1760
Labels
enhancement
New feature or request
With
lightning-thunder/thunder/core/trace.py
Line 671 in 621dce7
TraceCtx.python_callable()
call gets affected by the function according tolightning-thunder/thunder/core/trace.py
Lines 495 to 508 in 621dce7
From my point of view, the number of traces we'd want to modify would be 1 or 2: forward execution trace and/or backward execution trace.
Thus I think it'd be helpful to have an option to only make
set_execution_callback_file
effective to the two traces mentioned above.The text was updated successfully, but these errors were encountered: