1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
#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 c10d::control_plane
|
