mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
c4d1ff02f8
All changes other than the one to `tools/linter/adapters/s3_init_config.json` are generated by newer clang-format Pull Request resolved: https://github.com/pytorch/pytorch/pull/153889 Approved by: https://github.com/cyyever, https://github.com/atalman
45 lines
1.1 KiB
C++
45 lines
1.1 KiB
C++
#include <torch/csrc/distributed/c10d/control_plane/Handlers.hpp>
|
|
|
|
#include <cstdio>
|
|
#include <fstream>
|
|
#include <string>
|
|
|
|
#include <c10/util/tempfile.h>
|
|
#include <torch/csrc/distributed/c10d/exception.h>
|
|
#include <torch/csrc/utils/pybind.h>
|
|
|
|
namespace c10d::control_plane {
|
|
namespace {
|
|
|
|
RegisterHandler tracebackHandler{
|
|
"dump_traceback",
|
|
[](const Request&, Response& res) {
|
|
auto tmpfile = c10::make_tempfile("torch-dump_traceback");
|
|
|
|
auto cfile = ::fopen(tmpfile.name.c_str(), "w");
|
|
if (!cfile) {
|
|
throw std::runtime_error("failed to open file for writing");
|
|
}
|
|
|
|
{
|
|
py::gil_scoped_acquire guard{};
|
|
|
|
auto faulthandler = py::module::import("faulthandler");
|
|
faulthandler.attr("dump_traceback")(fileno(cfile), true);
|
|
}
|
|
|
|
::fclose(cfile);
|
|
|
|
std::ifstream file(tmpfile.name);
|
|
std::string str;
|
|
std::string file_contents;
|
|
while (std::getline(file, str)) {
|
|
file_contents += str;
|
|
file_contents.push_back('\n');
|
|
}
|
|
|
|
res.setContent(std::move(file_contents), "text/plain");
|
|
}};
|
|
} // namespace
|
|
} // namespace c10d::control_plane
|