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
|
#include "Halide.h"
using namespace Halide;
int main(int argc, char **argv) {
Target t = get_jit_target_from_environment().with_feature(Target::StrictFloat);
Var x;
ImageParam input(Float(32), 1);
Param<float> f_param;
Buffer<float> input_buffer(1);
input_buffer.fill(2.5f);
Func output;
output(x) = input(x + cast<int>(f_param));
input.set(input_buffer);
f_param.set(0.0f);
// This test verifies that this realize() doesn't explode in bounds infererence
// with "unbounded access of input"
Buffer<float> result = output.realize({1}, t);
assert(result(0) == 2.5f);
printf("Success!\n");
return 0;
}
|