1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
#include "Halide.h"
#include <stdio.h>
using namespace Halide;
int main(int argc, char **argv) {
Func f, g;
Var x, y;
RDom r(2, 18);
f(x, y) = 1;
f(r, y) = f(r - 2, y) + f(r - 1, y);
g(x, y) = f(x + 10, y) + 2;
// Provide estimates for pipeline output
g.set_estimates({{0, 50}, {0, 50}});
// Partially specify some schedules
g.reorder(y, x);
// Auto schedule the pipeline
Target target = get_target_from_environment();
Pipeline p(g);
// This should throw an error since auto-scheduler does not currently
// support partial schedules
p.apply_autoscheduler(target, {"Mullapudi2016"});
printf("Success!\n");
return 0;
}
|
