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 DiscRematerializationPass to auto recompute #1304

Closed
wants to merge 2 commits into from

Conversation

eedalong
Copy link
Collaborator

No description provided.

@eedalong eedalong requested a review from Yancey1989 June 19, 2024 08:33
@eedalong eedalong force-pushed the disc_remat branch 14 times, most recently from 21e1aa0 to e401c6d Compare June 26, 2024 10:09
@eedalong eedalong changed the title [WIP] Add DiscRematerializationPass to reduce peak memory Add DiscRematerializationPass to auto recompute Jun 26, 2024
@eedalong eedalong force-pushed the disc_remat branch 6 times, most recently from 29cb406 to d09f200 Compare June 27, 2024 06:59
@eedalong eedalong force-pushed the disc_remat branch 6 times, most recently from 6b45eb4 to 02cd54e Compare August 8, 2024 06:58
@eedalong
Copy link
Collaborator Author

This pr is too large, close it and will have several small prs for op scheduling and recompute for dynamic shape & static shapes seperately.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant