[BUG] LossModule
/TensorDictSequential
/ProbabilisticActor
cannot be compiled with torch.compile
#2312
Open
3 tasks done
Labels
bug
Something isn't working
Describe the bug
Attempting to invoke
torch.compile
on any of the abovementioned classes results in similar errors (see below)To Reproduce
Steps to reproduce the behavior.
Please try to provide a minimal example to reproduce the bug. Error messages and stack traces are also helpful.
Please use the markdown code blocks for both code and stack traces.
Expected behavior
Since non-compile version executes successfully, compile is expected to succeed.
System info
Describe the characteristic of your environment:
Reason and Possible fixes
Perhaps it is due to the decorators you mentioned in discord?
Checklist
The text was updated successfully, but these errors were encountered: