frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 21:49:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

Move functional collectives to the right namespace (#97793)

Browse Source 
This moves them from `torch._C._nn` to `torch._C._dist`
Pull Request resolved: https://github.com/pytorch/pytorch/pull/97793
Approved by: https://github.com/albanD
This commit is contained in:
Rodrigo Kumpera
2023-03-30 22:18:09 +00:00
committed by PyTorch MergeBot
parent 45acfc8574
commit 184bfbc3d7
11 changed files with 108 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/csrc/Module.cpp
View File

@ -41,6 +41,7 @@
#include <torch/csrc/TypeInfo.h>
#include <torch/csrc/api/include/torch/python/init.h>
#include <torch/csrc/autograd/python_cpp_function.h>
#include <torch/csrc/autograd/python_dist_functions.h>
#include <torch/csrc/autograd/python_enum_tag.h>
#include <torch/csrc/autograd/python_fft_functions.h>
#include <torch/csrc/autograd/python_function.h>
@ -1328,6 +1329,7 @@ PyObject* initModule() {
torch::autograd::initNestedFunctions(module);
torch::autograd::initSparseFunctions(module);
torch::autograd::initSpecialFunctions(module);
torch::autograd::initDistFunctions(module);
torch::autograd::init_legacy_variable(module);
torch::profiler::initPythonBindings(module);
torch::python::init_bindings(module);