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

Jump to bottom

fix layers/schedule for hybrid parallelization (#111) #112

Merged
merged 1 commit into from
Jan 4, 2022
Merged

fix layers/schedule for hybrid parallelization (#111) #112

merged 1 commit into from
Jan 4, 2022

Conversation

ver217
Copy link
Member

@ver217 ver217 commented Jan 4, 2022

  1. Pipeline schedule need re-compute batch size if using tensor parallel
  2. GPTBlock use checkpoint
  3. Some layer will raise exception if not using bias
  4. Add register_parameter for PipelineSharedModuleWrapper

@ver217 ver217 requested a review from kurisusnowdeng January 4, 2022 12:35
@ver217 ver217 added bug Something isn't working enhancement New feature or request labels Jan 4, 2022
@kurisusnowdeng kurisusnowdeng merged commit 7904baf into main Jan 4, 2022
@kurisusnowdeng kurisusnowdeng deleted the hotfix/hybrid-parallel branch January 4, 2022 12:52
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@kurisusnowdeng kurisusnowdeng kurisusnowdeng approved these changes

Assignees
No one assigned
Labels
bug Something isn't working enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@ver217 @kurisusnowdeng