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
|
# import halide as hl
# import numpy as np
def test_extern():
"""
Shows an example of Halide calling a C library loaded
in the Python process via ctypes
"""
# Requires Makefile support to build the external function in linkable form
print("[SKIP] TODO: test_extern not yet implemented in Python")
"""
data = np.random.random(10).astype(np.float64)
expected_result = np.sort(data)
output_data = np.empty(10, dtype=np.float64)
sort_func = hl.Func("extern_sort_func")
# gsl_sort,
# see http://www.gnu.org/software/gsl/manual/html_node/Sorting-vectors.html#Sorting-vectors
input = hl.ImageParam(hl.Float(64), 1, "input_data")
extern_name = "the_sort_func"
params = [hl.ExternFuncArgument(input)]
output_types = [hl.Int(32)]
dimensionality = 1
sort_func.define_extern(extern_name, params, output_types, dimensionality)
try:
sort_func.compile_jit()
except hl.HalideError as e:
assert "cannot be converted to a bool" in str(e)
else:
assert False, "Did not see expected exception!"
import ctypes
sort_lib = ctypes.CDLL("the_sort_function.so")
print(sort_lib.the_sort_func)
try:
sort_func.compile_jit()
except hl.HalideError as e:
assert "cannot be converted to a bool" in str(e)
else:
assert False, "Did not see expected exception!"
lib_path = "the_sort_function.so"
load_error = load_library_into_llvm(lib_path)
assert not load_error
sort_func.compile_jit()
# now that things are loaded, we try to call them
input.set(data)
sort_func.realize(output_data)
assert np.isclose(expected_result, output_data)
"""
if __name__ == "__main__":
test_extern()
|
