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

Fix compile wrapper #5455

Merged
merged 5 commits into from
May 8, 2024
Merged

Fix compile wrapper #5455

merged 5 commits into from
May 8, 2024

Conversation

BacharL
Copy link
Contributor

@BacharL BacharL commented Apr 23, 2024

compile wrapper will inherit from user module class and copy it's dict

This should resolve most issues in #5383 except potential extra user forward hooks.

@tohtana @loadams

compile wrapper will inherit from user module base class and copy it's __dict__

Change-Id: Ia21acf29da1f34593187c2119482d49059c4608a
@tjruwase tjruwase requested review from tohtana and umchand and removed request for duli2012 and mrwyattii April 24, 2024 15:40
@nelyahu
Copy link
Contributor

nelyahu commented May 8, 2024

@tohtana @tjruwase can you review this PR? this solves a major bug in current compile wrapper solution

@tohtana
Copy link
Contributor

tohtana commented May 8, 2024

Thank you for an awesome solution to the issues!

@tohtana tohtana enabled auto-merge May 8, 2024 08:50
@tohtana
Copy link
Contributor

tohtana commented May 8, 2024 via email

@tohtana tohtana added this pull request to the merge queue May 8, 2024
Merged via the queue into microsoft:master with commit 0b224ed May 8, 2024
14 checks passed
umchand pushed a commit to umchand/DeepSpeed that referenced this pull request May 20, 2024
compile wrapper will inherit from user module class and copy it's
__dict__

This should resolve most issues in microsoft#5383 except potential extra user
forward hooks.

@tohtana @loadams

Co-authored-by: Logan Adams <[email protected]>
Co-authored-by: Olatunji Ruwase <[email protected]>
Co-authored-by: Masahiro Tanaka <[email protected]>
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.

None yet

5 participants