Skip to content

Conversation

yf225
Copy link
Contributor

@yf225 yf225 commented Jun 7, 2018

This PR aims to be a better fix for #8246 by initializing the CUDA test flags in common_cuda only once, to avoid the duplicated import of TEST_CUDNN -> CUDA OOM issue on Windows.

We need to loop the test multiple times in CI to make sure there is no CUDA OOM error from this change.

@yf225 yf225 requested a review from apaszke June 7, 2018 20:44
@ezyang
Copy link
Contributor

ezyang commented Jun 7, 2018

@pytorchbot retest this please

@yf225 yf225 changed the title [WIP] Initialize CUDA test flags in common_cuda only once [DON'T RETEST] [WIP] Initialize CUDA test flags in common_cuda only once Jun 7, 2018
@ezyang
Copy link
Contributor

ezyang commented Jun 9, 2018

@pytorchbot retest this please

cls.__instance = object.__new__(cls)
return cls.__instance

def __init__(self):

This comment was marked as off-topic.

This comment was marked as off-topic.

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.

2 participants