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

[habana] Stub getMemInfo #2688

Merged
merged 1 commit into from
Apr 12, 2019
Merged

Conversation

bertmaher
Copy link
Contributor

Summary:
I took this idea from @jackm321, we need to actually estimate
memory or the host manager doesn't work.

Differential Revision: D14816916

Copy link
Contributor

@jackm321 jackm321 left a comment

Choose a reason for hiding this comment

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

LGTM

@@ -100,6 +96,26 @@ llvm::Error HabanaDeviceManager::init() {
return llvm::Error::success();
}

llvm::Error HabanaDeviceManager::updateMemoryUsage() {
// TODO: use synGetMemInfo once implemented
Copy link
Contributor

Choose a reason for hiding this comment

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

Period.

Period.

Summary:
I took this idea from @jackmontgomery, we need to actually estimate
memory or the host manager doesn't work.

Differential Revision: D14816916
@bertmaher bertmaher merged commit 14cb190 into pytorch:master Apr 12, 2019
@bertmaher bertmaher deleted the D14816916-6284d1b5 branch April 12, 2019 16:27
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.

4 participants