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