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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
#include "Halide.h"
// This test demonstrates using tracing to give you something like a
// stack trace in case of a crash (due to a compiler bug, or a bug in
// external code). We use a posix signal handler, which is probably
// os-dependent, so I'm going to enable this test on linux only.
#if defined(__linux__) || defined(__APPLE__) || defined(__unix) || defined(__posix)
#include <signal.h>
#include <stack>
#include <stdio.h>
#include <string>
using namespace Halide;
namespace {
using std::stack;
using std::string;
stack<string> stack_trace;
int my_trace(JITUserContext *user_context, const halide_trace_event_t *e) {
const string event_types[] = {"Load ",
"Store ",
"Begin realization ",
"End realization ",
"Produce ",
"Consume ",
"End consume ",
"Begin pipeline ",
"End pipeline "};
if (e->event == halide_trace_end_realization ||
e->event == halide_trace_consume ||
e->event == halide_trace_end_consume ||
e->event == halide_trace_end_pipeline) {
// These events signal the end of some previous event
stack_trace.pop();
}
if (e->event == halide_trace_begin_realization ||
e->event == halide_trace_produce ||
e->event == halide_trace_consume ||
e->event == halide_trace_begin_pipeline) {
// These events signal the start of some new region
stack_trace.push(event_types[e->event] + e->func);
}
return 0;
}
void signal_handler(int signum) {
printf("Correctly triggered a segfault. Here is the stack trace:\n");
while (!stack_trace.empty()) {
printf("%s\n", stack_trace.top().c_str());
stack_trace.pop();
}
printf("Success!\n");
exit(0);
}
} // namespace
int main(int argc, char **argv) {
#ifdef HALIDE_INTERNAL_USING_ASAN
// ASAN also needs to intercept the SIGSEGV signal handler;
// we could probably make these work together, but it's
// also probably not worth the effort.
printf("[SKIP] tracing_stack does not run under ASAN.\n");
return 0;
#endif
signal(SIGSEGV, signal_handler);
signal(SIGBUS, signal_handler);
// Loads from this image will barf, because we've messed up the host pointer
Buffer<int> input(100, 100);
halide_buffer_t *buf = input.raw_buffer();
buf->host = (uint8_t *)17;
Func f("f"), g("g"), h("h");
Var x("x"), y("y");
f(x, y) = x + y;
f.compute_root().trace_realizations();
g(x, y) = f(x, y) + 37;
g.compute_root().trace_realizations();
h(x, y) = g(x, y) + input(x, y);
h.trace_realizations();
h.jit_handlers().custom_trace = &my_trace;
h.realize({100, 100});
printf("The code should not have reached this print statement.\n");
return 1;
}
#else
#include <stdio.h>
int main(int argc, char **argv) {
printf("[SKIP] Test requires UNIX signal handling\n");
return 0;
}
#endif
|
