root/test/opengl/multiple_stages.cpp

DEFINITIONS

1. main

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

#include "testing.h"

using namespace Halide;

int main() {

    // This test must be run with an OpenGL target.
    const Target target = get_jit_target_from_environment().with_feature(Target::OpenGL);

    Func f, g, h;
    Var x, y, c;
    g(x, y, c) = cast<uint8_t>(x);
    h(x, y, c) = 1 + g(x, y, c);
    f(x, y, c) = h(x, y, c) + cast<uint8_t>(y);
    f.bound(c, 0, 3).glsl(x, y, c);
    h.bound(c, 0, 3).compute_root();
    g.bound(c, 0, 3).compute_root().glsl(x, y, c);

    Buffer<uint8_t> result = f.realize(10, 10, 3, target);
    result.copy_to_host();

    if (!Testing::check_result<uint8_t>(result, [&](int i, int j, int k) { return i + j + 1; })) {
        return 1;
    }

    Func f2, g2;
    f2(x, y, c) = cast<float>(x);
    g2(x, y, c) = f2(x, y, c) + cast<float>(y);

    f2.bound(c, 0, 3).glsl(x, y, c).compute_root();
    g2.bound(c, 0, 3).glsl(x, y, c);

    Buffer<float> result2 = g2.realize(10, 10, 3, target);
    if (!Testing::check_result<float>(result2, 0.01f, [&](int i, int j, int k) { return (float)(i + j); })) {
        return 1;
    }

    printf("Success!\n");

    return 0;
}