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

Published 20 hours ago • verified publisher icon Lightning-AI

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

Conservatively errors out when any of tensor inputs are a traceable tensor subclass and any non-pytorch executor is enabled.

I think it would be good to just error out and not depend on the executors here. WDYT?

t-vi avatar Oct 08 '24 06:10 t-vi