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
|
#include "Halide.h"
#include <assert.h>
#include <stdio.h>
using namespace Halide;
int error_occurred = false;
void my_error(JITUserContext *ctx, const char *msg) {
printf("Expected: %s\n", msg);
error_occurred = true;
}
int main(int argc, char **argv) {
Func f("f"), g("g"), h("h");
Var x("x"), y("y"), c("c");
RDom r(1, 99, "r");
g(x, y, c) = 42;
h(x, y, c) = 88;
f(x, y, c) = g(x, y, c);
f(r.x, r.y, c) = f(r.x - 1, r.y, c) + h(r.x, r.y, c);
f.jit_handlers().custom_error = my_error;
Buffer<int32_t> result = f.realize({100, 5, 3});
assert(error_occurred);
printf("Success!\n");
return 0;
}
|
