[go: nahoru, domu]
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

Make PjRtCompile accept XlaComputation by value and release XlaComputation as soon as it is converted into HloModule #72721

Draft
wants to merge 1 commit into
base: master
Choose a base branch
from
Draft

Make PjRtCompile accept XlaComputation by value and release XlaComputation as soon as it is converted into HloModule #72721

wants to merge 1 commit into from

Conversation

copybara-service[bot]
Copy link

Make PjRtCompile accept XlaComputation by value and release XlaComputation as soon as it is converted into HloModule

This avoids holding memory used by HloModuleProto during the entire compilation, which is often problematic because HloModuleProto sometimes uses much more memory than HloModule.

@junwhanahn @tensorflower-gardener
Make PjRtCompile accept XlaComputation by value and release `XlaC…
ccd4035 
…omputation` as soon as it is converted into `HloModule`

This avoids holding memory used by `HloModuleProto` during the entire compilation, which is often problematic because `HloModuleProto` sometimes uses much more memory than `HloModule`.

PiperOrigin-RevId: 657287204
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

None yet
1 participant
@junwhanahn