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
85
86
87
88
89
90
91
92
93
|
#===============================================================================
# Copyright 2020-2021 Intel Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#===============================================================================
# Parses kernel names from OpenCL files and updates KER_LIST_EXTERN and
# KER_LIST_ENTRIES variables with the parsed kernel names
function(parse_kernels ker_name ker_path)
set(entries "${KER_LIST_ENTRIES}")
file(READ ${ker_path} contents)
string(REGEX MATCHALL
"kernel[ \n]+void[ \n]+([a-z0-9_]+)"
kernels ${contents})
set(cur_ker_names)
foreach(k ${kernels})
string(REGEX REPLACE ".*void[ \n]+" "" k ${k})
list(APPEND cur_ker_names ${k})
list(FIND unique_ker_names ${k} index)
if (${index} GREATER -1)
message(WARNING "Kernel name is not unique: ${k}")
endif()
set(entries "${entries}\n { \"${k}\", ${ker_name}_kernel },")
endforeach()
set(KER_LIST_EXTERN
"${KER_LIST_EXTERN}\nextern const char *${ker_name}_kernel;"
PARENT_SCOPE)
set(KER_LIST_ENTRIES "${entries}" PARENT_SCOPE)
set(unique_ker_names "${unique_ker_names};${cur_ker_names}"
PARENT_SCOPE)
endfunction()
function(gen_gpu_kernel_list ker_list_templ ker_list_src ker_sources headers)
set(_sources "${SOURCES}")
set(KER_LIST_EXTERN)
set(KER_LIST_ENTRIES)
set(KER_HEADERS_EXTERN)
set(KER_HEADERS)
set(KER_HEADER_NAMES)
foreach(header_path ${headers})
get_filename_component(header_name ${header_path} NAME_WE)
string(REGEX REPLACE ".*\\/src\\/(.*)" "\\1" header_full_name ${header_path})
set(gen_file "${CMAKE_CURRENT_BINARY_DIR}/${header_name}_header.cpp")
add_custom_command(
OUTPUT ${gen_file}
COMMAND ${CMAKE_COMMAND}
-DCL_FILE="${header_path}"
-DGEN_FILE="${gen_file}"
-P ${PROJECT_SOURCE_DIR}/cmake/gen_gpu_kernel.cmake
DEPENDS ${header_path}
)
list(APPEND _sources "${gen_file}")
set(KER_HEADERS_EXTERN
"${KER_HEADERS_EXTERN}\nextern const char *${header_name}_header;")
set(KER_HEADER_LIST_ENTRIES
"${KER_HEADER_LIST_ENTRIES}\n {\"${header_full_name}\", ${header_name}_header},")
endforeach()
set(unique_ker_names)
foreach(ker_path ${ker_sources})
get_filename_component(ker_name ${ker_path} NAME_WE)
set(gen_file "${CMAKE_CURRENT_BINARY_DIR}/${ker_name}_kernel.cpp")
add_custom_command(
OUTPUT ${gen_file}
COMMAND ${CMAKE_COMMAND}
-DCL_FILE="${ker_path}"
-DGEN_FILE="${gen_file}"
-P ${PROJECT_SOURCE_DIR}/cmake/gen_gpu_kernel.cmake
DEPENDS ${ker_path}
)
list(APPEND _sources "${gen_file}")
parse_kernels(${ker_name} ${ker_path})
endforeach()
configure_file("${ker_list_templ}" "${ker_list_src}")
set(SOURCES "${_sources}" PARENT_SCOPE)
endfunction()
|
