lightning-thunder icon indicating copy to clipboard operation
lightning-thunder copied to clipboard

Disallow traceable tensor subclasses

Open crcrpar opened this issue 1 year ago • 1 comments

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.

crcrpar avatar Oct 08 '24 04:10 crcrpar