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
|
#include "Halide.h"
#include <stdio.h>
using namespace Halide;
int my_trace(JITUserContext *user_context, const halide_trace_event_t *e) {
if (e->event == halide_trace_store) {
for (int i = 0; i < e->type.lanes; ++i) {
int val = ((const int *)(e->value))[i];
if (val != 1234567890) {
printf("All values stored should have been 1234567890\n"
"Instead they are: %d\n",
val);
exit(1);
}
}
}
return 0;
}
int main(int argc, char **argv) {
Func f("f");
Var x("x"), y("y");
f(x, y) = 1234567890;
f.vectorize(x, 8);
f.trace_stores();
f.jit_handlers().custom_trace = &my_trace;
f.realize({8, 8});
printf("Success!\n");
return 0;
}
|
