root/test/opengl/copy_to_device.cpp

DEFINITIONS

1. main

#include "Halide.h"

#include "testing.h"

using namespace Halide;

// Test that internal allocations work correctly with copy_to_device.
// This requires that suitable buffer_t objects are created internally.
int main() {
    // This test must be run with an OpenGL target.
    const Target target = get_jit_target_from_environment().with_feature(Target::OpenGL);

    Buffer<uint8_t> input(255, 10, 3);
    input.fill([](int x, int y, int c) {
        return 10 * x + y + c;
    });

    Var x, y, c;
    Func g, h;
    h(x, y, c) = input(x, y, c);
    h.compute_root();  // force internal allocation of h

    // access h from shader to trigger copy_to_device operation
    g(x, y, c) = h(x, y, c);
    g.bound(c, 0, 3);
    g.glsl(x, y, c);

    Buffer<uint8_t> out(255, 10, 3);
    g.realize(out, target);
    out.copy_to_host();

    if (!Testing::check_result<uint8_t>(out, [&](int x, int y, int c) { return input(x, y, c); })) {
        return 1;
    }

    printf("Success!\n");
    return 0;
}