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
56
57
58
59
60
61
62
63
64
65
66
67
68
|
#include "Halide.h"
#include <iostream>
using namespace Halide;
int main(int argc, char **argv) {
// Test primary use case for unsafe_promise_clamped -- data that
// uses fewer bits than the type it is embeded within.
{
Buffer<uint16_t> ten_bit_data(100);
for (int i = 0; i < 100; i++) {
ten_bit_data(i) = i * 10;
}
Buffer<float> ten_bit_lut(1024);
for (int i = 0; i < 1024; i++) {
ten_bit_lut(i) = sin(2 * 3.1415f * i / 1024.0f);
}
Var x;
Func f;
ImageParam in(UInt(16), 1);
ImageParam lut(Float(32), 1);
f(x) = lut(unsafe_promise_clamped(in(x), 0, 1023));
lut.dim(0).set_bounds(0, 1024);
in.set(ten_bit_data);
lut.set(ten_bit_lut);
auto result = f.realize({100});
}
{
ImageParam in(UInt(8), 1);
ImageParam lut(Float(32), 1);
Var x;
Func f;
f(x) = lut(unsafe_promise_clamped(in(x), Expr(), 99));
f.infer_input_bounds({10});
Buffer<float> lut_bounds = lut.get();
assert(lut_bounds.dim(0).min() == 0 && lut_bounds.dim(0).extent() == 100);
}
{
ImageParam in(UInt(8), 1);
ImageParam lut(Float(32), 1);
Var x;
Func f;
f(x) = lut(unsafe_promise_clamped(in(x), 10, Expr()));
f.infer_input_bounds({10});
Buffer<float> lut_bounds = lut.get();
assert(lut_bounds.dim(0).min() == 10 && lut_bounds.dim(0).extent() == 246);
}
std::cout << "Success!\n";
return 0;
}
|
