To juggle CUDA streams from C++ I need to call THCState_getCurrentStreamOnDevice and friends. It would be nice to have an ATen wrapper for this to not have to include files from TH*.

The ATen include tree doesn't expose anything CUDA specific though, so some discussion is warranted before adding a wrapper. What would be the right place such a wrapper?

@ezyang @goldsborough @zdevito