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

Optionally use default-allocated GPU memory for long-lived buffers #1970

Merged
merged 5 commits into from
Sep 24, 2021

Conversation

timmoon10
Copy link
Collaborator

@timmoon10 timmoon10 commented Sep 18, 2021

This is a variant of PR #1964 that provides an option to disable that memory optimization.

Bamboo is running. Closes #1964.

Also affects GPU SGD and Adam buffers
@bvanessen
Copy link
Collaborator

@timmoon10 So just to be clear, the defaults on this will be to use the direct allocation. This seems reasonable, but having the flag lets us easily do performance evaluation.

@timmoon10 timmoon10 merged commit 2ac9c02 into LLNL:develop Sep 24, 2021
@timmoon10 timmoon10 deleted the resnet-memory-optimization2 branch September 24, 2021 18:42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants