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
|
#include "Halide.h"
#include <memory>
#include <stdio.h>
using namespace Halide;
int error_occurred = false;
void halide_error(JITUserContext *ctx, const char *msg) {
// Emitting "error.*:" to stdout or stderr will cause CMake to report the
// test as a failure on Windows, regardless of error code returned,
// hence the abbreviation to "err".
printf("Saw (Expected) Halide Err: %s\n", msg);
error_occurred = true;
}
static void test(int vector_width) {
Target target = get_jit_target_from_environment();
const int32_t kPrime1 = 7829;
const int32_t kPrime2 = 7919;
int32_t realize_width = vector_width ? vector_width : 1;
Buffer<int32_t> result;
Param<int32_t> p1, p2;
Var x;
Func s, f;
s(x) = p1 + p2;
f(x) = require(s(x) == kPrime1,
s(x) * kPrime2 + x,
"The parameters should add to exactly", kPrime1, "but were", s(x), "for vector_width", vector_width);
if (vector_width) {
s.vectorize(x, vector_width).compute_root();
f.vectorize(x, vector_width);
}
if (target.has_feature(Target::HVX)) {
f.hexagon();
}
f.jit_handlers().custom_error = halide_error;
// choose values that will fail
p1.set(1);
p2.set(2);
error_occurred = false;
result = f.realize({realize_width});
if (!error_occurred) {
printf("There should have been a requirement error (vector_width = %d)\n", vector_width);
exit(1);
}
p1.set(1);
p2.set(kPrime1 - 1);
error_occurred = false;
result = f.realize({realize_width});
if (error_occurred) {
printf("There should not have been a requirement error (vector_width = %d)\n", vector_width);
exit(1);
}
for (int i = 0; i < realize_width; ++i) {
const int32_t expected = (kPrime1 * kPrime2) + i;
const int32_t actual = result(i);
if (actual != expected) {
printf("Unexpected value at %d: actual=%d, expected=%d (vector_width = %d)\n", i, actual, expected, vector_width);
exit(1);
}
}
ImageParam input(Int(32), 2);
Expr h = require(p1 == p2, p1);
Func clamped = BoundaryConditions::repeat_edge(input, {{0, 64}, {0, h}});
clamped.jit_handlers().custom_error = &halide_error;
Buffer<int32_t> input_buf(64, 64);
input_buf.fill(0);
input.set(input_buf);
p1.set(16);
p2.set(15);
error_occurred = false;
result = clamped.realize({64, 3});
if (!error_occurred) {
printf("There should have been a requirement error (vector_width = %d)\n", vector_width);
exit(1);
}
p1.set(16);
p2.set(16);
error_occurred = false;
result = clamped.realize({64, 3});
if (error_occurred) {
printf("There should NOT have been a requirement error (vector_width = %d)\n", vector_width);
exit(1);
}
}
int main(int argc, char **argv) {
test(0);
test(4);
test(32);
printf("Success!\n");
return 0;
}
|
