Nikitha Malgi 12b73fdbbf Adding JIT support for cuda streams and events (#48020) ... Summary: ======= This PR addresses the following: * Adds JIT support for CUDA Streams * Adds JIT support for CUDA Events * Adds JIT support for CUDA Stream context manager Testing: ====== python test/test_jit.py -v TestCUDA Pull Request resolved: https://github.com/pytorch/pytorch/pull/48020 Reviewed By: navahgar Differential Revision: D25725749 Pulled By: nikithamalgifb fbshipit-source-id: b0addeb49630f8f0c430ed7badeca43bb9d2535c		2020-12-29 20:24:57 -08:00
..
api	Fix auto exponent issue for torch.pow (#49809)	2020-12-29 17:02:56 -08:00
common
dist_autograd	Fix Windows build failure after DDP PR merged (#45335)	2020-09-25 12:37:50 -07:00
jit	Adding JIT support for cuda streams and events (#48020)	2020-12-29 20:24:57 -08:00
rpc	[c10d] switch ProcessGroup to be managed by intrusive_ptr (#47343)	2020-11-12 07:36:23 -08:00
tensorexpr	[TensorExpr] Move SimpleIREval implementation from .h to .cpp. (#49697)	2020-12-21 20:20:15 -08:00
__init__.py