Revision 16ddff55efc02d37b713eed569d435bdc4f5dfb7 authored by Andrew Adams on 31 August 2023, 22:21:03 UTC, committed by Andrew Adams on 31 August 2023, 22:21:03 UTC
1 parent ef9a7d8
print_loop_nest.cpp
#include "Halide.h"
#include <stdio.h>
using namespace Halide;
int main(int argc, char **argv) {
{
Func output_y, output_u, output_v;
Var x, y, x_outer, y_outer, x_inner, y_inner;
Buffer<int> input(960, 960, 3);
output_y(x, y) = input(x, y, 0);
output_u(x, y) = input(2 * x, 2 * y, 1);
output_v(x, y) = input(2 * x, 2 * y, 2);
output_u.tile(x, y, x_outer, y_outer, x_inner, y_inner, 4, 1);
output_v.tile(x, y, x_outer, y_outer, x_inner, y_inner, 4, 1);
output_y.tile(x, y, x_outer, y_outer, x_inner, y_inner, 4 * 2, 2);
output_u.compute_with(output_y, x_outer);
output_v.compute_with(output_y, x_outer);
Pipeline p({output_y, output_u, output_v});
p.print_loop_nest();
}
printf("Success!\n");
return 0;
}
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...