torch/csrc/cuda/python_comm.h

#pragma once

namespace torch {
namespace cuda {
namespace python {

void initCommMethods(PyObject* module);

}
} // namespace cuda
} // namespace torch