inline void _autotune_timing_stub(Halide::Func& func) {
func.compile_jit();
// TODO: this assumes scalar/non-Tuple outputs - should generalize to a Realization
Halide::Type out_type = func.output_types()[0];
buffer_t out_size_buf;
{
// Use the Buffer constructor as a helper to set up the buffer_t,
// but then throw away its allocation which we don't really want.
Halide::Buffer bufinit(out_type, AUTOTUNE_N);
out_size_buf = *bufinit.raw_buffer();
out_size_buf.host = NULL;
}
Halide::Buffer out_size(out_type, &out_size_buf);
assert(out_size.host_ptr() == NULL); // make sure we don't have an allocation
func.infer_input_bounds(out_size);
// allocate the real output using the inferred mins + extents
Halide::Buffer output( out_type,
out_size.extent(0),
out_size.extent(1),
out_size.extent(2),
out_size.extent(3),
NULL,
"output" );
output.set_min( out_size.min(0),
out_size.min(1),
out_size.min(2),
out_size.min(3) );
// re-run input inference on enlarged output buffer
func.unbind_image_params(); // TODO: iterate to convergence
func.infer_input_bounds(output);
timeval t1, t2;
double rv = 0;
const unsigned int timeout = AUTOTUNE_LIMIT;
alarm(timeout);
for (int i = 0; i < AUTOTUNE_TRIALS; i++) {
gettimeofday(&t1, NULL);
func.realize(output);
gettimeofday(&t2, NULL);
alarm(0); // disable alarm
double t = (t2.tv_sec - t1.tv_sec) + (t2.tv_usec - t1.tv_usec)/1000000.0;
if(i == 0 || t < rv)
rv = t;
}
printf("{\"time\": %.10f}\n", rv);
exit(0);
}
inline void _autotune_timing_stub(Halide::Func& func) {
func.compile_jit();
func.infer_input_bounds(1024,1024);
timeval t1, t2;
double rv = 0;
for (int i = 0; i < 3; i++) {
gettimeofday(&t1, NULL);
func.realize(1024,1024);
gettimeofday(&t2, NULL);
double t = (t2.tv_sec - t1.tv_sec) + (t2.tv_usec - t1.tv_usec)/1000000.0;
if(i == 0 || t < rv)
rv = t;
}
printf("{\"time\": %.10f}\n", rv);
exit(0);
}
inline void _autotune_timing_stub(Halide::Func& func) {
func.compile_jit();
func.infer_input_bounds(AUTOTUNE_N);
timeval t1, t2;
double rv = 0;
const unsigned int timeout = AUTOTUNE_LIMIT;
alarm(timeout);
for (int i = 0; i < AUTOTUNE_TRIALS; i++) {
gettimeofday(&t1, NULL);
func.realize(AUTOTUNE_N);
gettimeofday(&t2, NULL);
alarm(0); // disable alarm
double t = (t2.tv_sec - t1.tv_sec) + (t2.tv_usec - t1.tv_usec)/1000000.0;
if(i == 0 || t < rv)
rv = t;
}
printf("{\"time\": %.10f}\n", rv);
exit(0);
}
void func_compile_jit1(h::Func &that, const h::Target &target = h::get_target_from_environment())
{
that.compile_jit(target);
return;
}
void func_compile_jit0(h::Func &that)
{
that.compile_jit();
return;
}
