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
torch_test_cpp_extension
complex_registration_extension.cpp	Replace open registration TensorTypeId with closed enum. (#25252)	2019-08-29 08:55:58 -07:00
cpp_frontend_extension.cpp
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
cuda_extension_kernel.cu
cuda_extension_kernel2.cu
cudnn_extension.cpp
doubler.h
extension.cpp
jit_extension.cpp
jit_extension2.cpp
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