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 an option to be able to disable replace_uniform #1660

Merged
merged 1 commit into from
Jan 19, 2025

Conversation

IvanYashchuk
Copy link
Collaborator

@IvanYashchuk IvanYashchuk commented Jan 18, 2025

Currently, Thunder decomposes Dropout into recomputation operations and rematerializes it in the nvFuser fusions in backward computations.

For easier experimentation what would the performance look like without dropout recomputation an option to disable the pass is added in this PR.

It's one of the requests from #1658.

cc @t-vi

@t-vi
Copy link
Collaborator

t-vi commented Jan 18, 2025

What could be an UX alternative here? I think we should find better ways than setting options for everything.

@t-vi t-vi enabled auto-merge (squash) January 19, 2025 10:25
Copy link
Collaborator

@t-vi t-vi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I would really, really, really not want the options to be part of our user experience. Let us merge this now to unblock #1658 quickly, but if you actually need this, we need a better solution to activate.

@t-vi t-vi merged commit 7c8bd3a into main Jan 19, 2025
50 checks passed
@t-vi t-vi deleted the disable-replace-uniform-option branch January 19, 2025 10:26
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants