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
|
#include "Halide.h"
namespace {
class UserContext : public Halide::Generator<UserContext> {
public:
Input<Buffer<float, 2>> input{"input"};
Output<Buffer<float, 2>> output{"output"};
void generate() {
Var x, y;
Func g;
g(x, y) = input(x, y) * 2;
g.compute_root();
output(x, y) = g(x, y);
output.parallel(y);
trace_pipeline();
// This test won't work in the profiler, because the profiler
// insists on calling malloc with nullptr user context.
assert(!get_target().has_feature(Target::Profile));
}
};
} // namespace
HALIDE_REGISTER_GENERATOR(UserContext, user_context)
|
