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
|
#include "Halide.h"
#include <memory>
#include <stdio.h>
using namespace Halide;
int error_occurred = false;
void halide_error(JITUserContext *ctx, const char *msg) {
printf("Expected: %s\n", msg);
error_occurred = true;
}
int main(int argc, char **argv) {
uint8_t c[4096];
memset(c, 42, sizeof(c));
halide_dimension_t shape[] = {{0, 4096, 1},
{0, 4096, 0},
{0, 256, 0}};
Halide::Buffer<uint8_t> buf(c, 3, shape);
ImageParam input(UInt(8), 3);
input.set(buf);
Var x;
Func grand_total;
grand_total() = cast<uint64_t>(input(0, 0, 0) + input(input.dim(0).extent() - 1, input.dim(1).extent() - 1, input.dim(2).extent() - 1));
grand_total.jit_handlers().custom_error = halide_error;
Target t = get_jit_target_from_environment();
Buffer<uint64_t> result;
if (t.bits != 32) {
grand_total.compile_jit(t.with_feature(Target::LargeBuffers));
result = grand_total.realize();
assert(!error_occurred);
assert(result(0) == (uint64_t)84);
}
grand_total.compile_jit(t);
result = grand_total.realize();
assert(error_occurred);
printf("Success!\n");
}
|
