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
|
#include "Halide.h"
#include <memory>
#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) {
Var x, y, z;
RDom r(0, 4096, 0, 4096, 0, 256);
Func big;
big(x, y, z) = cast<uint8_t>(42);
big.jit_handlers().custom_error = my_error;
big.compute_root();
Func grand_total;
grand_total() = cast<uint8_t>(sum(big(r.x, r.y, r.z)));
grand_total.jit_handlers().custom_error = my_error;
Buffer<uint8_t> result = grand_total.realize();
assert(error_occurred);
printf("Success!\n");
return 0;
}
|
