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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
#include "HalideBuffer.h"
#include "HalideRuntime.h"
#include <assert.h>
#include <math.h>
#include <stdio.h>
#if defined(TEST_OPENCL)
#include "HalideRuntimeOpenCL.h"
#endif
#include "gpu_texture.h"
using namespace Halide::Runtime;
#if defined(TEST_OPENCL)
#if !defined(HALIDE_RUNTIME_OPENCL)
#error "TEST_OPENCL defined but HALIDE_RUNTIME_OPENCL not defined"
#endif
#endif
int main(int argc, char **argv) {
#if defined(TEST_OPENCL)
const auto *interface = halide_opencl_device_interface();
assert(interface->compute_capability != nullptr);
int major, minor;
int err = interface->compute_capability(nullptr, &major, &minor);
if (err != 0 || (major == 1 && minor < 2)) {
printf("[SKIP] OpenCl %d.%d is less than required 1.2.\n", major, minor);
return 0;
}
const int W = 32, H = 32;
Buffer<int, 2> input(W, H);
for (int y = 0; y < input.height(); y++) {
for (int x = 0; x < input.width(); x++) {
input(x, y) = x + y;
}
}
// Explicitly copy data to the GPU.
input.set_host_dirty();
Buffer<int, 2> output(W, H);
gpu_texture(input, output);
if (input.raw_buffer()->device_interface != halide_opencl_image_device_interface()) {
printf("Expected input to be copied to texture storage");
return 1;
}
if (output.raw_buffer()->device_interface != halide_opencl_image_device_interface()) {
printf("Expected output to be copied to texture storage");
return 1;
}
output.copy_to_host();
// Verify output.
for (int y = 0; y < H; y++) {
for (int x = 0; x < W; x++) {
if (input(x, y) * 2 != output(x, y)) {
printf("Error at %d, %d: %d != %d\n", x, y, input(x, y), output(x, y));
return 1;
}
}
}
printf("Success!\n");
#else
printf("[SKIP] No OpenCL target enabled.\n");
#endif
return 0;
}
|
