Skip to content

Add option to use CUDA memory leak testing as a context manager#11380

Closed
zou3519 wants to merge 2 commits intopytorch:masterfrom
zou3519:cudacheck-contextmanager
Closed

Add option to use CUDA memory leak testing as a context manager#11380
zou3519 wants to merge 2 commits intopytorch:masterfrom
zou3519:cudacheck-contextmanager

Conversation

@zou3519
Copy link
Contributor

@zou3519 zou3519 commented Sep 7, 2018

cc @ssnl

@zou3519 zou3519 force-pushed the cudacheck-contextmanager branch from 13ed2ec to 187309d Compare September 7, 2018 16:03
Copy link
Collaborator

@ssnl ssnl left a comment

Choose a reason for hiding this comment

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

LGTM! Have one minor suggestion.

from common_cuda import initialize_cuda_context_rng
initialize_cuda_context_rng()

def __enter__(self):

This comment was marked as off-topic.

Copy link
Contributor

@facebook-github-bot facebook-github-bot left a comment

Choose a reason for hiding this comment

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

zou3519 has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

@zou3519
Copy link
Contributor Author

zou3519 commented Sep 10, 2018

Tests look OK; I'm ignoring the CircleCi tests.

PenghuiCheng pushed a commit to PenghuiCheng/pytorch that referenced this pull request Sep 11, 2018
…rch#11380)

Summary:
cc SsnL
Pull Request resolved: pytorch#11380

Reviewed By: ezyang

Differential Revision: D9705877

Pulled By: zou3519

fbshipit-source-id: 02470c25236f57fa02f4ac9d7ed63d38a6355db2
@zou3519 zou3519 deleted the cudacheck-contextmanager branch September 13, 2018 15:35
@ezyang ezyang added the merged label Jun 26, 2019
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.

5 participants