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

No recompute layers + Opt tensor fuse for sharding #752

Merged
merged 10 commits into from
Sep 26, 2022
Merged

No recompute layers + Opt tensor fuse for sharding #752

merged 10 commits into from
Sep 26, 2022

Conversation

FeixLiu
Copy link
Contributor

@FeixLiu FeixLiu commented Sep 19, 2022 •
edited
Loading

1.3B single

no recompute layers speed Speed Gain Memory
None 14954 NA 37428
0, 1, 2, 3, 4, 5 15426 +3.2% 40516

oom if more no_recompute_layers

1.3B 8DP

no recompute layers speed Speed Gain Memory
None 113371 NA 38942
0, 1, 2, 3 115534 +1.9 40526

oom if more no_recompute_layers

@FeixLiu FeixLiu force-pushed the recompute_layers branch 5 times, most recently from b94d434 to 40c55a8 Compare September 22, 2022 00:45
@FeixLiu FeixLiu mentioned this pull request Sep 22, 2022
Closed
@FeixLiu FeixLiu changed the title No recompute layers No recompute layers + Opt tensor fuse for sharding Sep 26, 2022
ForFishes
ForFishes approved these changes Sep 26, 2022
View reviewed changes
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 8c5efa6 into PaddlePaddle:develop Sep 26, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@ForFishes ForFishes ForFishes approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@FeixLiu @ForFishes