You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Currently, we can't use python_callable if we want to retrace because it does half of transform for execution.
We should absolutely get a flag on python_callable for that because we have a workaround in Thunder quite often now (in thunder/core/transforms.py but also in the autograd.Function handling in jit_ext now).
(the ["output"] might not make this example trivial, but still).
This is another, more prototypical, example the commented line does not work, but it should with a flag (or maybe even by default, so only transform for execution does it differently).
Currently, we can't use python_callable if we want to retrace because it does half of transform for execution.
We should absolutely get a flag on python_callable for that because we have a workaround in Thunder quite often now (in thunder/core/transforms.py but also in the autograd.Function handling in jit_ext now).
(the ["output"] might not make this example trivial, but still).
lightning-thunder/thunder/core/transforms.py
Lines 1438 to 1445 in ea96657
cc @apaz-cli
The text was updated successfully, but these errors were encountered: