Mike Ruberry a024e1e091 Creates Torch-friendly Event class and adds Stream tracking to autograd (#25130) ... Summary: Resubmission of https://github.com/pytorch/pytorch/issues/23424 because previous PR was borked. Pull Request resolved: https://github.com/pytorch/pytorch/pull/25130 Test Plan: Two tests were added to cuda_stream_test for this functionality. Differential Revision: D17145538 Pulled By: mruberry fbshipit-source-id: 2546c5907c038412e03aa0d3328a972b0164c455		2019-09-01 12:37:52 -07:00
..
no_python_abi_suffix_test	Allow building libraries with setuptools that dont have abi suffix (#14130)	2018-11-27 17:35:53 -08:00
torch_test_cpp_extension	Add support for dotted names in CPP Extensions (#6986)	2018-04-29 18:10:03 +02:00
complex_registration_extension.cpp	Replace open registration TensorTypeId with closed enum. (#25252)	2019-08-29 08:55:58 -07:00
cpp_frontend_extension.cpp	Better tests/support for Python/C++ inter-op (#15193)	2018-12-14 08:42:10 -08:00
cuda_extension_kernel.cu	Improve CUDA extension support (#5324)	2018-02-23 10:15:30 -05:00
cuda_extension_kernel2.cu	Improve CUDA extension support (#5324)	2018-02-23 10:15:30 -05:00
cuda_extension.cpp	Deprecate tensor.data<T>(), and codemod tensor.data<T>() to tensor.data_ptr<T>() (#24886)	2019-08-21 20:11:24 -07:00
cuda_extension.cu	Improve CUDA extension support (#5324)	2018-02-23 10:15:30 -05:00
cudnn_extension.cpp	Use torch:: instead of at:: in all C++ APIs (#13523)	2018-11-06 14:32:25 -08:00
doubler.h	Use torch:: instead of at:: in all C++ APIs (#13523)	2018-11-06 14:32:25 -08:00
extension.cpp	Use torch:: instead of at:: in all C++ APIs (#13523)	2018-11-06 14:32:25 -08:00
jit_extension.cpp	Unify C++ API with C++ extensions (#11510)	2018-09-24 14:44:21 -07:00
jit_extension2.cpp	Unify C++ API with C++ extensions (#11510)	2018-09-24 14:44:21 -07:00
msnpu_extension.cpp	Creates Torch-friendly Event class and adds Stream tracking to autograd (#25130)	2019-09-01 12:37:52 -07:00
setup.py	Expose backend extensions to python	2019-02-01 11:00:18 -08:00