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
105
106
107
108
109
|
#include "HalideBuffer.h"
#include "HalideRuntime.h"
#include "multitarget.h"
#include <atomic>
#include <string>
#include <tuple>
using namespace Halide::Runtime;
void my_error_handler(void *user_context, const char *message) {
// Don't use the word "error": if CMake sees it in the output
// from an add_custom_command() on Windows, it can decide that
// the command failed, regardless of error code.
printf("Saw: (%s)\n", message);
}
std::pair<std::string, bool> get_env_variable(char const *env_var_name) {
if (env_var_name) {
size_t read = 0;
#ifdef _MSC_VER
char lvl[32];
if (getenv_s(&read, lvl, env_var_name) != 0) read = 0;
#else
char *lvl = getenv(env_var_name);
read = (lvl) ? 1 : 0;
#endif
if (read) {
return {std::string(lvl), true};
}
}
return {"", false};
}
bool use_noboundsquery_feature() {
auto [value, read] = get_env_variable("HL_MULTITARGET_TEST_USE_NOBOUNDSQUERY_FEATURE");
if (!read) {
return false;
}
return std::stoi(value) != 0;
}
static std::atomic<int> can_use_count{0};
int my_can_use_target_features(int count, const uint64_t *features) {
can_use_count += 1;
const int word = halide_target_feature_no_bounds_query / 64;
const int bit = halide_target_feature_no_bounds_query % 64;
if (features[word] & (1ULL << bit)) {
if (use_noboundsquery_feature()) {
return 1;
} else {
return 0;
}
}
return 1;
}
int main(int argc, char **argv) {
const int W = 32, H = 32;
Buffer<uint32_t, 2> output(W, H);
halide_set_error_handler(my_error_handler);
halide_set_custom_can_use_target_features(my_can_use_target_features);
if (HalideTest::multitarget(output) != 0) {
printf("Error at multitarget\n");
return -1;
}
// Verify output.
for (int y = 0; y < H; y++) {
for (int x = 0; x < W; x++) {
const uint32_t expected = use_noboundsquery_feature() ? 0xdeadbeef : 0xf00dcafe;
const uint32_t actual = output(x, y);
if (actual != expected) {
printf("Error at %d, %d: expected %x, got %x\n", x, y, expected, actual);
return -1;
}
}
}
// halide_can_use_target_features() should be called exactly once, with the
// result cached; call this a few more times to verify.
for (int i = 0; i < 10; ++i) {
if (HalideTest::multitarget(output) != 0) {
printf("Error at multitarget\n");
return -1;
}
}
if (can_use_count != 1) {
printf("Error: halide_can_use_target_features was called %d times!\n", (int)can_use_count);
return -1;
}
{
// Verify that the multitarget wrapper code propagates nonzero error
// results back to the caller properly.
Buffer<uint8_t, 2> bad_type(W, H);
int result = HalideTest::multitarget(bad_type);
if (result != halide_error_code_bad_type) {
printf("Error: expected to fail with halide_error_code_bad_type (%d) but actually got %d!\n", (int)halide_error_code_bad_type, result);
return -1;
}
}
printf("Saw %x for no_bounds_query=%d\n", output(0, 0), use_noboundsquery_feature());
printf("Success!\n");
return 0;
}
|
