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