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 LinearTosRGB : public Halide::Generator<LinearTosRGB> {
Input<Func> linear{"linear"};
Output<Func> srgb{"srgb"};
Var x{"x"}, y{"y"};
void generate() {
using Halide::_;
srgb(x, y, _) = select(linear(x, y, _) <= .0031308f,
linear(x, y, _) * 12.92f,
(1 + .055f) * pow(linear(x, y, _), 1.0f / 2.4f) - .055f);
}
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.
linear.set_estimate(linear.args()[0], 0, W)
.set_estimate(linear.args()[1], 0, H);
for (size_t i = 2; i < linear.args().size(); ++i) {
linear.set_estimate(linear.args()[i], 0, C);
}
srgb.set_estimate(x, 0, W)
.set_estimate(y, 0, H);
for (size_t i = 2; i < srgb.args().size(); ++i) {
srgb.set_estimate(srgb.args()[i], 0, C);
}
} else {
Var yi("yi");
srgb.split(y, y, yi, 8).parallel(y).vectorize(x, 8);
}
}
};
} // namespace
HALIDE_REGISTER_GENERATOR(LinearTosRGB, linear_to_srgb)
|
