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
|
#include <assert.h>
#include <atomic>
#include <math.h>
#include <stdio.h>
#include "HalideBuffer.h"
#include "HalideRuntime.h"
#include "user_context_insanity.h"
using namespace Halide::Runtime;
const int num_launcher_tasks = 1000;
// Zero-initialized by virtue of being a global
static std::atomic<bool> got_context[num_launcher_tasks] = {};
int32_t my_halide_trace(void *context, const halide_trace_event_t *e) {
std::atomic<bool> *bool_ptr = (std::atomic<bool> *)context;
*bool_ptr = true;
return 0;
}
int launcher_task(void *user_context, int index, uint8_t *closure) {
Buffer<float, 2> input(10, 10);
for (int y = 0; y < 10; y++) {
for (int x = 0; x < 10; x++) {
input(x, y) = (float)(x * y);
}
}
Buffer<float, 2> output(10, 10);
user_context_insanity(&got_context[index], input, output);
return 0;
}
int main(int argc, char **argv) {
halide_set_custom_trace(my_halide_trace);
// Hijack halide's runtime to run a bunch of instances of this function
// in parallel.
// Note that launcher_task() always returns zero, thus halide_do_par_for()
// should always return zero, but since this is a test, let's verify that.
int result = halide_do_par_for(nullptr, launcher_task, 0, num_launcher_tasks, nullptr);
assert(result == 0);
(void)result;
for (int i = 0; i < num_launcher_tasks; ++i) {
assert(got_context[i] == true);
}
printf("Success!\n");
return 0;
}
|
