pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-11-06 09:17:11 +08:00

Files

Victor Quach 5830f122f1 Add docstrings for save_on_cpu hooks (#62410 )

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/62410

This PR adds docstrings for CPU hooks introduced in #61928.

Also uncomments the warning about pinned memory in CUDA semantics docs.

Depends on: #62361.

For now docstrings are an orphan page at https://docs-preview.pytorch.org/62410/generated/torch.autograd.graph.set_save_on_cpu_hooks.html#torch-autograd-graph-set-save-on-cpu-hooks

Test Plan: Imported from OSS

Reviewed By: soulitzer

Differential Revision: D29990129

Pulled By: Varal7

fbshipit-source-id: 7a98eeee6a0abb11e2c2d9169cd1aa35ad7ba3f4

2021-08-03 17:53:45 -07:00

amp_examples.rst

Reference amp tutorial (recipe) from core amp docs (#44725 )

2020-09-16 11:37:58 -07:00

autograd.rst

clarify default value of requires_grad for tensors (#61038 )

2021-07-12 12:57:37 -07:00

broadcasting.rst

Fixes docs (#51439 )

2021-01-31 22:00:26 -08:00

cpu_threading_runtimes.svg

Update CPU threading doc (#33083 )