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 38 Packages Projects Releases Wiki Activity
fe3ecfe0cf
pytorch/docs
History
shibo 6aeb85add8 add checkpoint support for custom device (#99626) 
Fixes #ISSUE_NUMBER
1、add checkpoint support for custom device
2、add a device argument, I want to add a device="cuda" parameter to the func `forward` of `CheckpointFunction`, and I can specify the device type when using it, but the func `apply` of `torch.autograd.Function` does not support `kwargs`, so I added a variable named `_device`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/99626
Approved by: https://github.com/soulitzer
2023-05-04 00:23:42 +00:00
..
caffe2 [BE] Remove dependency on six and future (#94709) 2023-02-14 09:14:14 +00:00
cpp Replace master with main in links and docs/conf.py (#100176) 2023-05-02 18:20:32 +00:00
source add checkpoint support for custom device (#99626) 2023-05-04 00:23:42 +00:00
.gitignore
libtorch.rst Replace master with main in links and docs/conf.py (#100176) 2023-05-02 18:20:32 +00:00
make.bat
Makefile Doc for Canonical Aten and Prims IR (#90644) 2022-12-13 21:30:47 +00:00
README.md
requirements.txt [ONNX] Document ONNX diagnostics (#88371) 2022-11-16 19:21:46 +00:00

README.md

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