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
|
#include "Halide.h"
namespace {
struct sRGBToLinear : public Halide::Generator<sRGBToLinear> {
Input<Func> srgb{"srgb"};
Output<Func> linear{"linear"};
Var x{"x"}, y{"y"};
void generate() {
using Halide::_;
linear(x, y, _) = select(srgb(x, y, _) <= 0.04045f,
srgb(x, y, _) / 12.92f,
pow(((srgb(x, y, _) + .055f) / (1.0f + .055f)), 2.4f));
}
void schedule() {
if (using_autoscheduler()) {
const int W = 1536, H = 2560, C = 4;
// Wart: Input<Func> are defined with Vars we don't know.
// Might be x,y but might be _0,_1. Use the args() to work around.
srgb.set_estimate(srgb.args()[0], 0, W)
.set_estimate(srgb.args()[1], 0, H);
for (size_t i = 2; i < srgb.args().size(); ++i) {
srgb.set_estimate(srgb.args()[i], 0, C);
}
linear.set_estimate(x, 0, W)
.set_estimate(y, 0, H);
for (size_t i = 2; i < linear.args().size(); ++i) {
linear.set_estimate(linear.args()[i], 0, C);
}
} else {
Var yi("yi");
linear.split(y, y, yi, 8).parallel(y).vectorize(x, 8);
}
}
};
} // namespace
HALIDE_REGISTER_GENERATOR(sRGBToLinear, srgb_to_linear)
|
