Hi everyone,
I was wondering if there was a way to access the single-segment buffer interface of CUDA Tensors. For example, when using pyCUDA, I used to do:
def buff(ary):
    return ary.gpudata.as_buffer(ary.nbytes)

I’d like to use mpi4py with Pytorch, and while I can use the numpy interface for CPU tensors, this is not possible for GPU ones. This would be a nice trick while we wait for the distributed interface to be polished.
Thanks a lot for your help (and for developing Pytorch),
Séb