FM / cuda

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/cuda/

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/cuda

Name	Size	Mode	Actions
comm.h	1525	0644	edit dl rm
device_set.h	152	0644	edit dl rm
Event.h	487	0644	edit dl rm
Module.h	503	0644	edit dl rm
nccl.h	5037	0644	edit dl rm
override_macros.h	1887	0644	edit dl rm
python_comm.h	114	0644	edit dl rm
python_nccl.h	616	0644	edit dl rm
restore_macros.h	820	0644	edit dl rm
serialization.h	524	0644	edit dl rm
Storage.h	2585	0644	edit dl rm
Stream.h	524	0644	edit dl rm
THCP.h	421	0644	edit dl rm
undef_macros.h	932	0644	edit dl rm
utils.h	871	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/cuda/python_nccl.h (616B)

#pragma once #include 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);