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
|
#include "Halide.h"
#include <stdio.h>
#include <thread>
using namespace Halide;
int main(int argc, char **argv) {
// Wasm JIT is substantially slower than others,
// so do fewer iterations to avoid timing out.
const bool is_wasm = get_jit_target_from_environment().arch == Target::WebAssembly;
// Test if the compiler itself is thread-safe. This test is
// intended to be run in a thread-sanitizer.
// std::thread has implementation-dependent behavior; some implementations
// may refuse to create an arbitrary number. So let's create a smallish
// number (8) and have each one do enough work that contention is likely
// to be encountered.
const int total_iters = is_wasm ? 256 : 1024;
constexpr int num_threads = 8;
std::vector<std::thread> threads;
for (int i = 0; i < num_threads; i++) {
threads.emplace_back([=] {
for (int i = 0; i < (total_iters / num_threads); i++) {
Func f;
Var x;
f(x) = x;
f.realize({100});
}
});
}
for (auto &t : threads) {
t.join();
}
printf("Success!\n");
return 0;
}
|
