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.

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

Add option to ThunderCompiler to save gm.code or gm.print_readable to file #1082

Open
crcrpar opened this issue Sep 2, 2024 · 1 comment
Assignees
Labels
dynamo+thunder for things that could be applicable to the dynamo+thunder frontend enhancement New feature or request torch.compile

Comments

@crcrpar
Copy link
Collaborator

crcrpar commented Sep 2, 2024

🚀 Feature

Motivation

It'd be common to see torch.compile frontend generates multiple graphs for a model and Thunder fails to interpret and optimize them.
In such case, it'd be helpful if we have an option to save them into files.

Pitch

Alternatives

Additional context

cc @apaz-cli

@crcrpar crcrpar added enhancement New feature or request torch.compile labels Sep 2, 2024
@tfogal tfogal self-assigned this Sep 6, 2024
@tfogal
Copy link
Collaborator

tfogal commented Sep 6, 2024

Assigning to me, I've got some things here that I just need to clean up and make a PR out of

@tfogal tfogal added the dynamo+thunder for things that could be applicable to the dynamo+thunder frontend label Sep 6, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
dynamo+thunder for things that could be applicable to the dynamo+thunder frontend enhancement New feature or request torch.compile
Projects
None yet
Development

No branches or pull requests

2 participants