pytorch/torch/csrc/distributed/python_placement.h
Scott Wolchok 7d16fcf2df Re-re-re-re-apply "C++-accessible Placements via pybind11 (#163030)" (#166132)
Was reverted (again!) due to a merge conflict that crept in sometime during the "export to github -> land internally -> merge on github" process.

D85096233

Pull Request resolved: https://github.com/pytorch/pytorch/pull/166132
Approved by: https://github.com/Skylion007, https://github.com/ezyang, https://github.com/malfet
2025-10-27 21:19:32 +00:00

8 lines
168 B
C++

#pragma once
#include <torch/csrc/utils/python_stub.h>
namespace torch::distributed {
void initPlacementBindings(PyObject* module);
} // namespace torch::distributed