1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
#include <c10/macros/Macros.h>
#include <torch/csrc/jit/backends/backend_debug_info.h>
namespace torch::jit::backend {
namespace {
#ifdef BUILD_LITE_INTERPRETER
static auto cls = torch::class_<PyTorchBackendDebugInfoDummy>(
kBackendUtilsNamespace,
kBackendDebugInfoClass)
.def(torch::init<>());
#else
static auto cls = torch::class_<PyTorchBackendDebugInfo>(
kBackendUtilsNamespace,
kBackendDebugInfoClass)
.def(torch::init<>());
#endif
} // namespace
} // namespace torch::jit::backend
|
