projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 1e425d1) | patch
Move all Stream and Event Python implementation to C++ (#15937)
authorShen Li <shenli@fb.com>
Thu, 17 Jan 2019 15:22:42 +0000 (07:22 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 17 Jan 2019 15:29:22 +0000 (07:29 -0800)
commit24f4d3987ec643fe245608625cfbf8997c42e87e
tree5f5a30d0543877d67ce1eff2ffec63fc2bdadc79tree | snapshot
parent1e425d1a47ab6292a592120e335fd51ca56d12b3commit | diff
Move all Stream and Event Python implementation to C++ (#15937)

Summary:
1. Added `torch/csrc/cuda/Event.h` and `torch/csrc/cuda/Event.cpp` to bind Python Event class to C++ implementation.
2. Move all CUDA runtime invocations from `torch/cuda/streams.py` to C++
3. Added tests to cover Stream and Event APIs. ~(event IPC handle tests is introduced in #15974)~
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15937

Differential Revision: D13649001

Pulled By: mrshenli

fbshipit-source-id: 84ca58f35f6ba679a4ba33150ceba678d760d240
16 files changed:
aten/src/ATen/cuda/CUDAEvent.h diff | blob | history
aten/src/ATen/test/cuda_stream_test.cpp diff | blob | history
c10/cuda/CUDAStream.h diff | blob | history
test/test_cuda.py diff | blob | history
test/test_multiprocessing.py diff | blob | history
tools/autograd/templates/python_variable_methods.cpp diff | blob | history
torch/CMakeLists.txt diff | blob | history
torch/csrc/Module.cpp diff | blob | history
torch/csrc/cuda/Event.cpp [new file with mode: 0644] blob
torch/csrc/cuda/Event.h [new file with mode: 0644] blob
torch/csrc/cuda/Stream.cpp diff | blob | history
torch/csrc/cuda/Stream.h diff | blob | history
torch/csrc/cuda/THCP.h diff | blob | history
torch/cuda/__init__.py diff | blob | history
torch/cuda/streams.py diff | blob | history
torch/multiprocessing/reductions.py diff | blob | history
Domain: Machine Learning / ML Framework;