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
|
#include <stdio.h>
#include "HalideBuffer.h"
#include "halide_benchmark.h"
#include "shuffler.h"
using Halide::Runtime::Buffer;
using namespace Halide::Tools;
int main(int argc, char **argv) {
{
constexpr int W = 256;
Buffer<int32_t, 1> input(W);
for (int x = 0; x < W; x++) {
input(x) = x;
}
Buffer<int32_t, 1> output(W / 4);
shuffler(input, output);
for (int x = 0; x < W / 4; x++) {
int expected = input(input(x / 2 + 1) / 2 + 1) + 1;
int actual = output(x);
if (expected != actual) {
printf("at x = %d expected %d got %d\n", x, expected, actual);
return 1;
}
}
}
{
constexpr int W = 16384;
Buffer<int32_t, 1> input(W);
for (int x = 0; x < W; x++) {
input(x) = x;
}
Buffer<int32_t, 1> output(W / 4);
BenchmarkConfig config;
config.min_time = 1.0;
config.max_time = config.min_time * 4;
double best = benchmark([&]() {
shuffler(input, output);
},
config);
printf("Best time: %f\n", best);
}
printf("Success!\n");
return 0;
}
|
