Caffe2 - C++ API
A deep learning, cross platform ML framework
python_nccl.h
1 #pragma once
2 
3 #include <torch/csrc/python_headers.h>
4 
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);