lightning-thunder
lightning-thunder copied to clipboard
Disallow traceable tensor subclasses
What does this PR do?
Conservatively errors out when any of tensor inputs are a traceable tensor subclass and any non-pytorch executor is enabled.
Ideally we should interpret, translate, and disable __torch_dispatch__ eventually.