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
74
75
76
77
78
79
80
81
82
83
84
|
cmake_minimum_required(VERSION 3.28)
project(Halide_Python_apps)
if (PROJECT_IS_TOP_LEVEL)
enable_testing()
endif ()
find_package(Halide REQUIRED COMPONENTS Python)
if (Halide_TARGET MATCHES "wasm")
message(WARNING "Python apps are skipped under WASM.")
return()
endif ()
if (Halide_TARGET MATCHES "vulkan")
message(WARNING "Python apps are skipped under Vulkan.")
return()
endif ()
if (NOT WITH_AUTOSCHEDULERS)
message(WARNING "Python apps are skipped without autoschedulers")
return()
endif ()
list(APPEND CMAKE_MODULE_PATH "${CMAKE_CURRENT_SOURCE_DIR}/../cmake")
include(AddPythonTest)
set(TEST_TMPDIR "$<SHELL_PATH:${CMAKE_CURRENT_BINARY_DIR}>")
set(TEST_IMAGES_DIR "$<SHELL_PATH:${CMAKE_CURRENT_SOURCE_DIR}/../../apps/images>")
set(APPS
bilateral_grid
blur
identity
interpolate
local_laplacian)
set(GENERATORS_bilateral_grid bilateral_grid bilateral_grid_Adams2019 bilateral_grid_Li2018 bilateral_grid_Mullapudi2016)
set(GENERATORS_blur blur)
set(GENERATORS_identity identity)
set(GENERATORS_interpolate interpolate interpolate_Mullapudi2016)
set(GENERATORS_local_laplacian local_laplacian local_laplacian_Mullapudi2016)
set(ARGS_bilateral_grid ${TEST_IMAGES_DIR}/gray.png 0.1 ${TEST_TMPDIR}/out.png)
set(ARGS_blur ${TEST_IMAGES_DIR}/gray.png ${TEST_TMPDIR}/out.png)
set(ARGS_identity "")
set(ARGS_interpolate ${TEST_IMAGES_DIR}/rgba.png ${TEST_TMPDIR}/out.png)
set(ARGS_local_laplacian ${TEST_IMAGES_DIR}/rgba.png 8 1 1 ${TEST_TMPDIR}/out.png)
foreach (app IN LISTS APPS)
set(app_generator_src "${app}_generator.py")
add_halide_generator(app_gen_${app}
SOURCES ${app_generator_src})
set(DEPS "")
foreach (G IN ITEMS ${GENERATORS_${app}})
add_halide_library(app_aot_${G}
FROM app_gen_${app}
GENERATOR ${G}
FUNCTION_NAME ${G}
USE_RUNTIME ${RUNTIME_${G}}
PYTHON_EXTENSION _ignored_result
# We don't really need all the plugins at once here --
# it's just easier to specify them all
PLUGINS Halide::Adams2019 Halide::Li2018 Halide::Mullapudi2016)
add_halide_python_extension_library(app_ext_${G}
MODULE_NAME ${G}
HALIDE_LIBRARIES app_aot_${G})
list(APPEND DEPS app_ext_${G})
endforeach ()
set(app_src "${app}_app.py")
add_python_test(
FILE "${app_src}"
TEST_ARGS ${ARGS_${app}}
LABEL python_apps
DEPENDS ${DEPS}
PYTHONPATH "$<TARGET_FILE_DIR:app_ext_${app}>"
ENVIRONMENT
"TEST_TMPDIR=${TEST_TMPDIR}"
"TEST_IMAGES_DIR=${TEST_IMAGES_DIR}"
)
endforeach ()
|
