Skip to content

Add InternalStorage and add ShardingOptimizerStage2 #37489

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

Merged
merged 1 commit into from
Nov 25, 2021

Conversation

Baibaifan
Copy link
Contributor

@Baibaifan Baibaifan commented Nov 23, 2021

PR types

New features

PR changes

APIs

Describe

  1. Add InternalStorage method. Parameter memory can be integrated by parameter integration.
    origin param ->cpu -> integration->gpu
    image
  2. Add ShardingOptimizerStage2
    ShardingOptimizer encapsulates the optimization strategy and integrates it into the optimizer.

@paddle-bot-old
Copy link

paddle-bot-old bot commented Nov 23, 2021

✅ This PR's description meets the template requirements!
Please wait for other CI results.

@paddle-bot-old
Copy link

Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

Copy link
Contributor

@XieYunshen XieYunshen left a comment

Choose a reason for hiding this comment

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

LGTM for set_tests_properties(test_dygraph_sharding_optimizer_stage2 PROPERTIES TIMEOUT 120)

Copy link
Member

@ForFishes ForFishes left a comment

Choose a reason for hiding this comment

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

LGTM

@ForFishes ForFishes merged commit 5af6463 into PaddlePaddle:develop Nov 25, 2021
Zjq9409 pushed a commit to Zjq9409/Paddle that referenced this pull request Dec 10, 2021
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.

3 participants