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
|
#include "Halide.h"
namespace {
class Multitarget : public Halide::Generator<Multitarget> {
public:
Output<Buffer<uint32_t, 2>> output{"output"};
Output<Buffer<float, 0>> random_float_output{"random_float_output"};
Output<Buffer<int32_t, 0>> random_int_output{"random_int_output"};
void generate() {
Var x, y;
// Note that 'NoBoundsQuery' is essentially a somewhat-arbitrary placeholder
// here; we really just want to use a feature flag that doesn't require
// a custom runtime (as does, e.g., Target::Debug).
if (get_target().has_feature(Target::NoBoundsQuery)) {
output(x, y) = cast<uint32_t>((int32_t)0xdeadbeef);
} else {
output(x, y) = cast<uint32_t>((int32_t)0xf00dcafe);
}
random_float_output() = Halide::random_float();
random_int_output() = Halide::random_int();
}
};
} // namespace
HALIDE_REGISTER_GENERATOR(Multitarget, multitarget)
|
