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

[shardformer] fix pipeline forward error if custom layer distribution is used #5189

Merged
merged 9 commits into from
Mar 27, 2024
Merged

[shardformer] fix pipeline forward error if custom layer distribution is used #5189

merged 9 commits into from
Mar 27, 2024

Commits on Mar 27, 2024

  1. Use self.[distribute_layers|get_stage_index] to exploit custom layer … 

    …distribution
    @insujang @CWHer
    insujang authored and CWHer committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    20fa9a6 View commit details
    Browse the repository at this point in the history

  2. Change static methods for t5 layer distribution to member functions

    @insujang @CWHer
    insujang authored and CWHer committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    0d9a70e View commit details
    Browse the repository at this point in the history

  3. Change static methods for whisper layer distribution to member functions

    @insujang @CWHer
    insujang authored and CWHer committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    31f8365 View commit details
    Browse the repository at this point in the history

  4. Replace whisper policy usage with self one

    @insujang @CWHer
    insujang authored and CWHer committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    dd9d2c6 View commit details
    Browse the repository at this point in the history

  5. Fix test case to use non-static layer distribution methods

    @insujang @CWHer
    insujang authored and CWHer committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    485f874 View commit details
    Browse the repository at this point in the history

  6. Fix missing changes

    @insujang @CWHer
    insujang authored and CWHer committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    e8d4b7f View commit details
    Browse the repository at this point in the history

  7. style: polish code

    @CWHer
    CWHer committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    86c069e View commit details
    Browse the repository at this point in the history

  8. feat: Use self.[distribute_layers|get_stage_index] to exploit custom … 

    …layer distribution
    @CWHer
    CWHer committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    00f94fa View commit details
    Browse the repository at this point in the history

  9. fix: fix typo

    @CWHer
    CWHer committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    a74c702 View commit details
    Browse the repository at this point in the history