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
39
40
|
#include "Halide.h"
namespace {
class CleanupOnError : public Halide::Generator<CleanupOnError> {
public:
Output<Buffer<int32_t, 1>> output{"output"};
void generate() {
Var x;
// This allocation is going to succeed
Func f;
f(x) = x;
f.compute_root();
Target target = get_target();
if (target.has_gpu_feature() && !target.has_feature(Target::Metal)) {
// Skip Metal, because it uses zero-copy, which breaks the
// assumptions of the test.
Var xo, xi;
f.gpu_tile(x, xo, xi, 16);
}
// This one is going to fail (because we'll override
// halide_malloc to make it fail). The first allocation should
// be cleaned up when the second one fails.
Func g;
g(x) = f(2 * x) + f(2 * x + 1);
g.compute_root();
output(x) = g(x) + 1;
}
};
} // namespace
HALIDE_REGISTER_GENERATOR(CleanupOnError, cleanup_on_error)
|
