commit 12b73fdbbf4a89f3ec46983f90a2bb2d866cb338
author Nikitha Malgi <nikithamalgi@fb.com> Tue Dec 29 20:22:19 2020 -0800
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> Tue Dec 29 20:24:57 2020 -0800
tree 9454cc6b6675da4998a863cee46449251046c2a5
parent 97c17b47720dd1afce88180b843c9182f1a665a9

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

tools/build_variables.bzl

diff --git a/tools/build_variables.bzl b/tools/build_variables.bzl
index a214684..ec53f1d 100644
--- a/tools/build_variables.bzl
+++ b/tools/build_variables.bzl
@@ -408,6 +408,7 @@
     "torch/csrc/jit/codegen/cuda/transform_rfactor.cpp",
     "torch/csrc/jit/codegen/cuda/type.cpp",
     "torch/csrc/jit/tensorexpr/cuda_codegen.cpp",
+    "torch/csrc/jit/runtime/register_cuda_ops.cpp",
 ]
 
 libtorch_cuda_sources = libtorch_cuda_core_sources + [