Add PYTORCH_TEST_SKIP_CUDA_MEM_LEAK_CHECK (#1237)
Summary:
This PR allows to skip CUDA memory leak check by using `PYTORCH_TEST_SKIP_CUDA_MEM_LEAK_CHECK=1 python test.py`.
The same config is used in PyTorch unittests:
https://github.com/pytorch/pytorch/blob/a348975e00081334ac96d855932d2753a62f1e77/torch/testing/_internal/common_utils.py#L993
Pull Request resolved: https://github.com/pytorch/benchmark/pull/1237
Reviewed By: sangongs
Differential Revision: D40274579
Pulled By: xuzhao9
fbshipit-source-id: 42aeb7d43c0d08a1bbaf681fb3b690801995a529
