Skip to content

[DON'T RETEST] [WIP] Initialize CUDA test flags in common_cuda only once#8253

Closed
yf225 wants to merge 1 commit intopytorch:masterfrom
yf225:win_cuda_check
Closed

[DON'T RETEST] [WIP] Initialize CUDA test flags in common_cuda only once#8253
yf225 wants to merge 1 commit intopytorch:masterfrom
yf225:win_cuda_check

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