root/test/correctness/autotune_bug_4.cpp

/* [<][>][^][v][top][bottom][index][help] */

DEFINITIONS

This source file includes following definitions.
  1. my_trace
  2. main

#include "Halide.h"
#include <stdio.h>

using namespace Halide;

int my_trace(void *user_context, const halide_trace_event_t *e) {
    // The schedule implies that f and g will be stored from 0 to 7
    if (e->event == 2 && std::string(e->func) == "f") {
        if (e->coordinates[1] < 7) {
            printf("Bounds on realization were supposed to be = [0, 7]\n"
                   "Instead they are: %d %d\n", e->coordinates[0], e->coordinates[1]);
            exit(-1);
        }
    }
    return 0;
}

int main(int argc, char **argv) {
    Func f("f"), g("g"), h("h");
    Var x("x");

    f(x) = x;
    g(x) = f(x);
    h(x) = g(x) + g(x+1);

    Var xo("xo"), xi("xi");
    f.split(x, xo, xi, 4);
    g.split(x, xo, xi, 5);
    h.split(x, xo, xi, 6);
    f.compute_at(h, xo);
    g.compute_at(h, xo);
    g.store_root();

    f.trace_realizations().trace_stores().trace_loads();
    g.trace_realizations().trace_stores().trace_loads();

    h.set_custom_trace(&my_trace);
    h.bound(x, 0, 6);
    h.realize(6);

    printf("Success!\n");

    return 0;

}

/* [<][>][^][v][top][bottom][index][help] */