3 #include <torch/csrc/python_headers.h> 5 PyObject * THCPModule_nccl_version(PyObject *
self, PyObject *args);
6 PyObject * THCPModule_nccl_unique_id(PyObject *
self, PyObject *args);
7 PyObject * THCPModule_nccl_init_rank(PyObject *
self, PyObject *args);
8 PyObject * THCPModule_nccl_reduce(PyObject *
self, PyObject *args);
9 PyObject * THCPModule_nccl_all_reduce(PyObject *
self, PyObject *args);
10 PyObject * THCPModule_nccl_broadcast(PyObject *
self, PyObject *args);
11 PyObject * THCPModule_nccl_all_gather(PyObject *
self, PyObject *args);
12 PyObject * THCPModule_nccl_reduce_scatter(PyObject *
self, PyObject *args);