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
33
34
35
36
37
38
|
#include "Halide.h"
#include <stdio.h>
using namespace Halide;
#ifdef _WIN32
#define DLLEXPORT __declspec(dllexport)
#else
#define DLLEXPORT
#endif
int call_counter = 0;
extern "C" DLLEXPORT int count(int arg) {
call_counter++;
return arg;
}
HalideExtern_1(int, count, int);
int main(int argc, char **argv) {
Func f, g;
Var x;
g(x) = count(x);
f(x) = g(100 - x) + g(100 - x + 1);
g.compute_at(f, x);
g.store_root();
f.realize({10});
if (call_counter != 11) {
printf("g was called %d times instead of %d\n", call_counter, 11);
return -1;
}
printf("Success!\n");
return 0;
}
|
