OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 37 Packages Projects Releases Wiki Activity
5b767d404e
pytorch/docs
History
Codrin Popa 5b767d404e Modified roundup_power2_divisions to specify the number of divisions for each power of two interval (#87290) 
Summary:
Improved roundup_power2_divisions knob so it allows better control of rouding in the PyTorch CUDA Caching Allocator.

This new version allows setting the number of divisions per power of two interval starting from 1MB and ending at 64GB and above. An example use case is when rouding is desirable for small allocations but there are also very large allocations which are persistent, thus would not benefit from rounding and take up extra space.

Test Plan: Tested locally

Differential Revision: D40103909

Pull Request resolved: https://github.com/pytorch/pytorch/pull/87290
Approved by: https://github.com/zdevito
2022-11-04 19:31:16 +00:00
..
caffe2 Fix typos under docs directory (#88033) 2022-10-31 19:31:56 +00:00
cpp Fix typos under docs directory (#88033) 2022-10-31 19:31:56 +00:00
source Modified roundup_power2_divisions to specify the number of divisions for each power of two interval (#87290) 2022-11-04 19:31:16 +00:00
.gitignore
libtorch.rst
make.bat
Makefile [ONNX] Clean up onnx_supported_ops (#79424) 2022-06-23 20:44:51 +00:00
README.md Add docs/README.md to make existing doc build info more discoverable (#49286) 2020-12-16 11:55:45 -08:00
requirements.txt Freeze the rest of python docs requirement (#83785) 2022-08-20 03:37:21 +00:00

README.md

Please see the Writing documentation section of CONTRIBUTING.md for details on both writing and building the docs.