pytorch

1
#include <torch/csrc/utils/cpp_stacktraces.h>
2

3
#include <cstdlib>
4
#include <cstring>
5

6
#include <c10/util/Exception.h>
7

8
namespace torch {
9
namespace {
10
bool compute_cpp_stack_traces_enabled() {
11
  auto envar = std::getenv("TORCH_SHOW_CPP_STACKTRACES");
12
  if (envar) {
13
    if (strcmp(envar, "0") == 0) {
14
      return false;
15
    }
16
    if (strcmp(envar, "1") == 0) {
17
      return true;
18
    }
19
    TORCH_WARN(
20
        "ignoring invalid value for TORCH_SHOW_CPP_STACKTRACES: ",
21
        envar,
22
        " valid values are 0 or 1.");
23
  }
24
  return false;
25
}
26

27
bool compute_disable_addr2line() {
28
  auto envar = std::getenv("TORCH_DISABLE_ADDR2LINE");
29
  if (envar) {
30
    if (strcmp(envar, "0") == 0) {
31
      return false;
32
    }
33
    if (strcmp(envar, "1") == 0) {
34
      return true;
35
    }
36
    TORCH_WARN(
37
        "ignoring invalid value for TORCH_DISABLE_ADDR2LINE: ",
38
        envar,
39
        " valid values are 0 or 1.");
40
  }
41
  return false;
42
}
43
} // namespace
44

45
bool get_cpp_stacktraces_enabled() {
46
  static bool enabled = compute_cpp_stack_traces_enabled();
47
  return enabled;
48
}
49

50
bool get_disable_addr2line() {
51
  static bool disabled = compute_disable_addr2line();
52
  return disabled;
53
}
54

55
} // namespace torch
56