root/test/generator/nested_externs_aottest.cpp

DEFINITIONS

1. main

#include <stdio.h>
#include <stdlib.h>

#include "HalideBuffer.h"
#include "nested_externs_root.h"

using namespace Halide::Runtime;

int main(int argc, char **argv) {
    auto buf = Buffer<float>::make_interleaved(100, 200, 3);

    nested_externs_root(38.5f, buf);

    buf.for_each_element([&](int x, int y, int c) {
        const float correct = 158.0f;
        const float actual = buf(x, y, c);
        if (actual != correct) {
            printf("result(%d, %d, %d) = %f instead of %f\n",
                   x, y, c, actual, correct);
            exit(-1);
        }
    });

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