Nikitha Malgi 416ba5c48f Merge CUDA Streams and Events (#53902) ... Summary: ----------- - Updates current_stream and default stream API's to take `optional[device]` argument - Adds parsing logic to replace `torch.cuda.Stream` and `torch.cuda.Event` -> `torch.classes.cuda.Stream` and `torch.classes.cuda.Event` for JIT - Merges StreamContext manager for both Eager and JIT. Pull Request resolved: https://github.com/pytorch/pytorch/pull/53902 Test Plan: ------ Run JIT tests: python test/test_jit.py -v TestCUDA Run eager tests: python test/test_cuda.py -v TestCuda Reviewed By: SplitInfinity Differential Revision: D27285996 Pulled By: nikithamalgifb fbshipit-source-id: 45d9fee9a582b5f4c82330f5f99eb88584804270		2021-03-26 14:19:39 -07:00
..
api	Add padding='same' mode to conv{1,2,3}d (#45667)	2021-03-18 16:22:03 -07:00
common
dist_autograd	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
jit	Merge CUDA Streams and Events (#53902)	2021-03-26 14:19:39 -07:00
lite_interpreter_runtime	[PyTorch] Enable test_lite_interpreter_runtime running in android (#54579)	2021-03-24 14:45:27 -07:00
rpc	Use type-erased union for Buffer. (#54251)	2021-03-19 04:58:09 -07:00
tensorexpr	[TensorExpr] Add plumbing for conv2d fusion. (#54439)	2021-03-24 18:49:07 -07:00
__init__.py