Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Tensor Subclasses] Trace transfom to interpret __torch_dispatch__ #1394

Open
wants to merge 20 commits into
base: main
Choose a base branch
from
+1,088 −4

avoid flattening non-tensor args of subclass ctor

e9027ba
Select commit
Loading
Failed to load commit list.
Open

[Tensor Subclasses] Trace transfom to interpret __torch_dispatch__ #1394

avoid flattening non-tensor args of subclass ctor
e9027ba
Select commit
Loading
Failed to load commit list.
Azure Pipelines / lightning-thunder (ipynb) (jupyter ubuntu22.04 | cuda 12.1 | torch-nightly) succeeded Dec 21, 2024 in 8m 36s

jupyter ubuntu22.04 | cuda 12.1 | torch-nightly succeeded

0 errors / 0 warnings

View more details on Azure Pipelines