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
41
42
43
|
#include "Halide.h"
using namespace Halide;
// from issue #3221
int main(int argc, char *argv[]) {
Target t = get_jit_target_from_environment();
if (!t.has_feature(Target::CUDA)) {
printf("[SKIP] CUDA not enabled\n");
return 0;
}
Var x, y, p, d;
Func f1, f2;
f1(x, y, p) = 0;
f2(x, y, p) = 0;
RDom r(0, 10, 0, 10);
Func b1, b2;
b1(p) = 0.f;
b1(p) += f1(r.x, r.y, p);
b2(p) = 0.f;
b2(p) += f2(r.x, r.y, p);
Func d1, d2;
d1(p) = b1(p) - b2(p);
d2(p) = b2(p) - b1(p);
Func result;
result(d, p) = select(d == 0, d1(p), d2(p));
d2.compute_root().gpu_blocks(p);
// this used to cause an assertion error
result.compile_jit(Target("host-cuda"));
printf("Success!\n");
return 0;
}
|
