pytorch
774b638e - Change largeCUDATensorTest to largeTensorTest+onlyCUDA; add a buffer to large cuda tensor test (#45332)

Commit
4 years ago
Change largeCUDATensorTest to largeTensorTest+onlyCUDA; add a buffer to large cuda tensor test (#45332) Summary: Effectively, `largeCUDATensorTest` = `largeTensorTest` + `onlyCUDA`. There was this problem where a user got OOM for a `largeCUDATensorTest('16GB')` on a 16GB V100. This decorator was checking total memory for a GPU device, however in most cases, we can't allocate all of the memory that a GPU has. So, it would be beneficial that we have a buffer on this `largeTensorTest` check for CUDA. I added a 10% buffer to it. Definition of `largeTensorTest` https://github.com/pytorch/pytorch/blob/d22dd80128a0d1cfdbfc174a20271719b2f9c1e9/torch/testing/_internal/common_device_type.py#L560-L578 `_has_sufficient_memory` https://github.com/pytorch/pytorch/blob/d22dd80128a0d1cfdbfc174a20271719b2f9c1e9/torch/testing/_internal/common_device_type.py#L535-L557 `largeCUDATensorTest` https://github.com/pytorch/pytorch/blob/d22dd80128a0d1cfdbfc174a20271719b2f9c1e9/torch/testing/_internal/common_device_type.py#L526-L532 Pull Request resolved: https://github.com/pytorch/pytorch/pull/45332 Reviewed By: ngimel Differential Revision: D24698690 Pulled By: mruberry fbshipit-source-id: a77544478e45ce271f6639ea04e87700574ae307
Author
Parents
Loading