Why Gemfury? Push, build, and install  RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Bower components Debian packages RPM packages NuGet packages

neilisaac / torch   python

Repository URL to install this package:

Version: 1.8.0 

/ include / torch / csrc / cuda / python_nccl.h

#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);