[CUDA graphs] Prototype API and documentation (#63269)
authorMichael Carilli <mcarilli@gmail.com>
Tue, 31 Aug 2021 20:29:39 +0000 (13:29 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Tue, 31 Aug 2021 20:34:23 +0000 (13:34 -0700)
commit8d08b103be936d78d5d4ed90c0547aeccb8ce166
tree2654bc3bce78c9af2ccb4414992bfe849dce3774
parent1c2b5e59ae53ef3042ad1b5cf9aa275391c92971
[CUDA graphs] Prototype API and documentation (#63269)

Summary:
RFC: https://github.com/pytorch/pytorch/issues/61880

Pull Request resolved: https://github.com/pytorch/pytorch/pull/63269

Reviewed By: mruberry

Differential Revision: D30596643

Pulled By: ngimel

fbshipit-source-id: b1f8061406364b667e2c2d4d30fbce1f0d8456be
aten/src/ATen/cuda/CUDAGraphsUtils.cuh
aten/src/ATen/native/cudnn/Conv_v7.cpp
c10/cuda/CUDACachingAllocator.cpp
docs/source/cuda.rst
docs/source/notes/cuda.rst
test/test_cuda.py
torch/_C/__init__.pyi.in
torch/csrc/cuda/Graph.cpp
torch/cuda/__init__.py
torch/cuda/graphs.py [new file with mode: 0644]
torch/cuda/streams.py