generic.cmake 16.0 KB
Newer Older
L
liaogang 已提交
1
# Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.
2
#
L
liaogang 已提交
3 4 5
# 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
6
#
L
liaogang 已提交
7
# http://www.apache.org/licenses/LICENSE-2.0
8
#
L
liaogang 已提交
9 10 11 12 13
# 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.
14
#
15 16


Y
Yi Wang 已提交
17 18 19
# generic.cmake defines CMakes functions that look like Bazel's
# building rules (https://bazel.build/).
#
W
wuyi05 已提交
20
#
L
liaogang 已提交
21
# -------------------------------------------
22
#     C++        CUDA C++       Go
L
liaogang 已提交
23
# -------------------------------------------
24 25 26
# cc_library    nv_library   go_library
# cc_binary     nv_binary    go_binary
# cc_test       nv_test      go_test
L
liaogang 已提交
27
# -------------------------------------------
W
wuyi05 已提交
28
#
Y
Yi Wang 已提交
29 30
# To build a static library example.a from example.cc using the system
#  compiler (like GCC):
W
wuyi05 已提交
31
#
32
#   cc_library(example SRCS example.cc)
W
wuyi05 已提交
33
#
Y
Yi Wang 已提交
34 35
# To build a static library example.a from multiple source files
# example{1,2,3}.cc:
W
wuyi05 已提交
36
#
37
#   cc_library(example SRCS example1.cc example2.cc example3.cc)
W
wuyi05 已提交
38
#
39
# To build a shared library example.so from example.cc:
W
wuyi05 已提交
40
#
41
#   cc_library(example SHARED SRCS example.cc)
W
wuyi05 已提交
42
#
Y
Yi Wang 已提交
43 44
# To build a library using Nvidia's NVCC from .cu file(s), use the nv_
# prefixed version:
W
wuyi05 已提交
45
#
46
#   nv_library(example SRCS example.cu)
W
wuyi05 已提交
47
#
48
# To specify that a library new_example.a depends on other libraies:
W
wuyi05 已提交
49
#
50
#   cc_library(new_example SRCS new_example.cc DEPS example)
W
wuyi05 已提交
51
#
52
# Static libraries can be composed of other static libraries:
W
wuyi05 已提交
53
#
54
#   cc_library(composed DEPS dependent1 dependent2 dependent3)
W
wuyi05 已提交
55
#
Y
Yi Wang 已提交
56 57
# To build an executable binary file from some source files and
# dependent libraries:
W
wuyi05 已提交
58
#
59
#   cc_binary(example SRCS main.cc something.cc DEPS example1 example2)
W
wuyi05 已提交
60
#
Y
Yi Wang 已提交
61 62
# To build an executable binary file using NVCC, use the nv_ prefixed
# version:
W
wuyi05 已提交
63
#
64
#   nv_binary(example SRCS main.cc something.cu DEPS example1 example2)
W
wuyi05 已提交
65
#
Y
Yi Wang 已提交
66 67
# To build a unit test binary, which is an executable binary with
# GoogleTest linked:
W
wuyi05 已提交
68
#
69
#   cc_test(example_test SRCS example_test.cc DEPS example)
W
wuyi05 已提交
70
#
71
# To build a unit test binary using NVCC, use the nv_ prefixed version:
W
wuyi05 已提交
72
#
73
#   nv_test(example_test SRCS example_test.cu DEPS example)
74
#
Y
Yi Wang 已提交
75 76 77
# It is pretty often that executable and test binaries depend on
# pre-defined external libaries like glog and gflags defined in
# /cmake/external/*.cmake:
78
#
79
#   cc_test(example_test SRCS example_test.cc DEPS example glog gflags)
L
liaogang 已提交
80 81 82 83 84 85 86 87 88
#
# To build a go static library using Golang, use the go_ prefixed version:
#
#   go_library(example STATIC)
#
# To build a go shared library using Golang, use the go_ prefixed version:
#
#   go_library(example SHARED)
#
L
liaogang 已提交
89

Y
Yu Yang 已提交
90
# including binary directory for generated headers.
91
include_directories(${CMAKE_CURRENT_BINARY_DIR})
Y
Yu Yang 已提交
92

93
if(NOT APPLE AND NOT ANDROID)
L
liaogang 已提交
94 95
    find_package(Threads REQUIRED)
    link_libraries(${CMAKE_THREAD_LIBS_INIT})
G
gangliao 已提交
96
    set(CMAKE_CXX_LINK_EXECUTABLE "${CMAKE_CXX_LINK_EXECUTABLE} -ldl -lrt")
H
hedaoyuan 已提交
97
endif(NOT APPLE AND NOT ANDROID)
98

99
function(merge_static_libs TARGET_NAME)
L
liaogang 已提交
100 101
  set(libs ${ARGN})
  list(REMOVE_DUPLICATES libs)
102

103
  # Get all propagation dependencies from the merged libraries
L
liaogang 已提交
104
  foreach(lib ${libs})
105 106
    list(APPEND libs_deps ${${lib}_LIB_DEPENDS})
  endforeach()
L
liaogang 已提交
107
  list(REMOVE_DUPLICATES libs_deps)
108

109 110 111 112
  # To produce a library we need at least one source file.
  # It is created by add_custom_command below and will helps
  # also help to track dependencies.
  set(target_SRCS ${CMAKE_CURRENT_BINARY_DIR}/${TARGET_NAME}_dummy.c)
113

114
  if(APPLE) # Use OSX's libtool to merge archives
115 116 117
    # Make the generated dummy source file depended on all static input
    # libs. If input lib changes,the source file is touched
    # which causes the desired effect (relink).
118 119
    add_custom_command(OUTPUT ${target_SRCS}
      COMMAND ${CMAKE_COMMAND} -E touch ${target_SRCS}
120 121 122
      DEPENDS ${libs})

    # Generate dummy staic lib
123 124
    file(WRITE ${target_SRCS} "const char *dummy = \"${target_SRCS}\";")
    add_library(${TARGET_NAME} STATIC ${target_SRCS})
125 126 127 128 129 130
    target_link_libraries(${TARGET_NAME} ${libs_deps})

    foreach(lib ${libs})
      # Get the file names of the libraries to be merged
      set(libfiles ${libfiles} $<TARGET_FILE:${lib}>)
    endforeach()
L
liaogang 已提交
131
    add_custom_command(TARGET ${TARGET_NAME} POST_BUILD
132
      COMMAND rm "${CMAKE_CURRENT_BINARY_DIR}/lib${TARGET_NAME}.a"
133 134
      COMMAND /usr/bin/libtool -static -o "${CMAKE_CURRENT_BINARY_DIR}/lib${TARGET_NAME}.a" ${libfiles}
      )
135
  else() # general UNIX: use "ar" to extract objects and re-add to a common lib
136 137
    set(target_DIR ${CMAKE_CURRENT_BINARY_DIR}/${TARGET_NAME}.dir)

L
liaogang 已提交
138
    foreach(lib ${libs})
139 140
      set(objlistfile ${target_DIR}/${lib}.objlist) # list of objects in the input library
      set(objdir ${target_DIR}/${lib}.objdir)
141

L
liaogang 已提交
142
      add_custom_command(OUTPUT ${objdir}
143 144
        COMMAND ${CMAKE_COMMAND} -E make_directory ${objdir}
        DEPENDS ${lib})
145

L
liaogang 已提交
146 147
      add_custom_command(OUTPUT ${objlistfile}
        COMMAND ${CMAKE_AR} -x "$<TARGET_FILE:${lib}>"
148
        COMMAND ${CMAKE_AR} -t "$<TARGET_FILE:${lib}>" > ${objlistfile}
L
liaogang 已提交
149 150
        DEPENDS ${lib} ${objdir}
        WORKING_DIRECTORY ${objdir})
151

152
      list(APPEND target_OBJS "${objlistfile}")
L
liaogang 已提交
153
    endforeach()
154

155 156 157 158 159 160 161 162 163 164
    # Make the generated dummy source file depended on all static input
    # libs. If input lib changes,the source file is touched
    # which causes the desired effect (relink).
    add_custom_command(OUTPUT ${target_SRCS}
      COMMAND ${CMAKE_COMMAND} -E touch ${target_SRCS}
      DEPENDS ${libs} ${target_OBJS})

    # Generate dummy staic lib
    file(WRITE ${target_SRCS} "const char *dummy = \"${target_SRCS}\";")
    add_library(${TARGET_NAME} STATIC ${target_SRCS})
165
    target_link_libraries(${TARGET_NAME} ${libs_deps})
166 167

    # Get the file name of the generated library
168
    set(target_LIBNAME "$<TARGET_FILE:${TARGET_NAME}>")
169

170 171 172 173
    add_custom_command(TARGET ${TARGET_NAME} POST_BUILD
        COMMAND ${CMAKE_AR} crs ${target_LIBNAME} `find ${target_DIR} -name '*.o'`
        COMMAND ${CMAKE_RANLIB} ${target_LIBNAME}
        WORKING_DIRECTORY ${target_DIR})
L
liaogang 已提交
174
  endif()
175 176
endfunction(merge_static_libs)

L
liaogang 已提交
177
function(cc_library TARGET_NAME)
178
  set(options STATIC static SHARED shared)
L
liaogang 已提交
179 180 181
  set(oneValueArgs "")
  set(multiValueArgs SRCS DEPS)
  cmake_parse_arguments(cc_library "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
182 183 184 185 186 187 188 189
  if (cc_library_SRCS)
    if (cc_library_SHARED OR cc_library_shared) # build *.so
      add_library(${TARGET_NAME} SHARED ${cc_library_SRCS})
    else()
      add_library(${TARGET_NAME} STATIC ${cc_library_SRCS})
    endif()
    if (cc_library_DEPS)
      add_dependencies(${TARGET_NAME} ${cc_library_DEPS})
190
      target_link_libraries(${TARGET_NAME} ${cc_library_DEPS})
191
    endif()
L
liaogang 已提交
192 193
    
    # cpplint code style
L
liaogang 已提交
194 195 196 197 198 199 200
    foreach(source_file ${cc_library_SRCS})
      string(REGEX REPLACE "\\.[^.]*$" "" source ${source_file})
      if(EXISTS ${CMAKE_CURRENT_SOURCE_DIR}/${source}.h)
        list(APPEND cc_library_HEADERS ${CMAKE_CURRENT_SOURCE_DIR}/${source}.h)
      endif()
    endforeach()
    add_style_check_target(${TARGET_NAME} ${cc_library_SRCS} ${cc_library_HEADERS})
L
liaogang 已提交
201

202
  else(cc_library_SRCS)
203
    if(cc_library_DEPS)
204 205 206 207 208
      merge_static_libs(${TARGET_NAME} ${cc_library_DEPS})
    else()
      message(FATAL "Please specify source file or library in cc_library.")
    endif()
  endif(cc_library_SRCS)
L
liaogang 已提交
209 210
endfunction(cc_library)

L
liaogang 已提交
211
function(cc_binary TARGET_NAME)
212
  set(options "")
L
liaogang 已提交
213 214 215 216
  set(oneValueArgs "")
  set(multiValueArgs SRCS DEPS)
  cmake_parse_arguments(cc_binary "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
  add_executable(${TARGET_NAME} ${cc_binary_SRCS})
217
  if(cc_binary_DEPS)
L
liaogang 已提交
218 219 220
    target_link_libraries(${TARGET_NAME} ${cc_binary_DEPS})
    add_dependencies(${TARGET_NAME} ${cc_binary_DEPS})
  endif()
L
liaogang 已提交
221 222
endfunction(cc_binary)

L
liaogang 已提交
223
function(cc_test TARGET_NAME)
224 225 226 227 228 229
  if(WITH_TESTING)
    set(options "")
    set(oneValueArgs "")
    set(multiValueArgs SRCS DEPS)
    cmake_parse_arguments(cc_test "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
    add_executable(${TARGET_NAME} ${cc_test_SRCS})
L
liaogang 已提交
230 231
    target_link_libraries(${TARGET_NAME} ${cc_test_DEPS} gtest gtest_main)
    add_dependencies(${TARGET_NAME} ${cc_test_DEPS} gtest gtest_main)
G
gongweibao 已提交
232
    add_test(NAME ${TARGET_NAME} COMMAND ${TARGET_NAME} WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR})
L
liaogang 已提交
233
  endif()
L
liaogang 已提交
234
endfunction(cc_test)
L
liaogang 已提交
235 236

function(nv_library TARGET_NAME)
237
  if (WITH_GPU)
238
    set(options STATIC static SHARED shared)
239 240 241
    set(oneValueArgs "")
    set(multiValueArgs SRCS DEPS)
    cmake_parse_arguments(nv_library "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
242 243 244 245 246 247 248 249
    if(nv_library_SRCS)
      if (nv_library_SHARED OR nv_library_shared) # build *.so
        cuda_add_library(${TARGET_NAME} SHARED ${nv_library_SRCS})
      else()
          cuda_add_library(${TARGET_NAME} STATIC ${nv_library_SRCS})
      endif()
      if (nv_library_DEPS)
        add_dependencies(${TARGET_NAME} ${nv_library_DEPS})
250
        target_link_libraries(${TARGET_NAME} ${nv_library_DEPS})
251
      endif()
L
liaogang 已提交
252 253 254 255
      # cpplint code style
      foreach(source_file ${nv_library_SRCS})
        string(REGEX REPLACE "\\.[^.]*$" "" source ${source_file})
        if(EXISTS ${CMAKE_CURRENT_SOURCE_DIR}/${source}.h)
Q
Qiao Longfei 已提交
256
          list(APPEND nv_library_HEADERS ${CMAKE_CURRENT_SOURCE_DIR}/${source}.h)
L
liaogang 已提交
257 258 259
        endif()
      endforeach()
      add_style_check_target(${TARGET_NAME} ${nv_library_SRCS} ${nv_library_HEADERS})
260 261 262 263 264 265 266
    else(nv_library_SRCS)
      if (nv_library_DEPS)
        merge_static_libs(${TARGET_NAME} ${nv_library_DEPS})
      else()
        message(FATAL "Please specify source file or library in nv_library.")
      endif()
    endif(nv_library_SRCS)
267
  endif()
L
liaogang 已提交
268 269 270
endfunction(nv_library)

function(nv_binary TARGET_NAME)
271 272 273 274 275 276 277 278 279 280
  if (WITH_GPU)
    set(options "")
    set(oneValueArgs "")
    set(multiValueArgs SRCS DEPS)
    cmake_parse_arguments(nv_binary "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
    cuda_add_executable(${TARGET_NAME} ${nv_binary_SRCS})
    if(nv_binary_DEPS)
      target_link_libraries(${TARGET_NAME} ${nv_binary_DEPS})
      add_dependencies(${TARGET_NAME} ${nv_binary_DEPS})
    endif()
L
liaogang 已提交
281
  endif()
L
liaogang 已提交
282 283 284
endfunction(nv_binary)

function(nv_test TARGET_NAME)
285 286 287 288 289 290
  if (WITH_GPU AND WITH_TESTING)
    set(options "")
    set(oneValueArgs "")
    set(multiValueArgs SRCS DEPS)
    cmake_parse_arguments(nv_test "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
    cuda_add_executable(${TARGET_NAME} ${nv_test_SRCS})
L
liaogang 已提交
291 292
    target_link_libraries(${TARGET_NAME} ${nv_test_DEPS} gtest gtest_main)
    add_dependencies(${TARGET_NAME} ${nv_test_DEPS} gtest gtest_main)
293
    add_test(${TARGET_NAME} ${TARGET_NAME})
L
liaogang 已提交
294
  endif()
L
liaogang 已提交
295
endfunction(nv_test)
L
liaogang 已提交
296 297

function(go_library TARGET_NAME)
L
liaogang 已提交
298
  set(options STATIC static SHARED shared)
L
liaogang 已提交
299
  set(oneValueArgs "")
L
liaogang 已提交
300
  set(multiValueArgs DEPS)
L
liaogang 已提交
301
  cmake_parse_arguments(go_library "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
L
liaogang 已提交
302 303

  if (go_library_SHARED OR go_library_shared)
L
liaogang 已提交
304
    set(BUILD_MODE "-buildmode=c-shared")
H
Helin Wang 已提交
305
    set(${TARGET_NAME}_LIB_NAME "${CMAKE_SHARED_LIBRARY_PREFIX}${TARGET_NAME}${CMAKE_SHARED_LIBRARY_SUFFIX}" CACHE STRING "output library name for target ${TARGET_NAME}")
L
liaogang 已提交
306 307
  else()
    set(BUILD_MODE "-buildmode=c-archive")
H
Helin Wang 已提交
308
    set(${TARGET_NAME}_LIB_NAME "${CMAKE_STATIC_LIBRARY_PREFIX}${TARGET_NAME}${CMAKE_STATIC_LIBRARY_SUFFIX}" CACHE STRING "output library name for target ${TARGET_NAME}")
L
liaogang 已提交
309
  endif()
L
liaogang 已提交
310 311

  set(dummyfile ${CMAKE_CURRENT_BINARY_DIR}/${TARGET_NAME}_dummy.c)
312 313 314 315 316 317 318 319 320 321 322 323 324 325 326

  # This custom command will always run since it depends on a not
  # existing file.
  add_custom_command(
    OUTPUT dummy_rebulid_${TARGET_NAME}
    COMMAND cmake -E touch ${dummyfile}
    )
  # Create a custom target that depends on the custom command output
  # file, so the custom command can be referenced as a dependency by
  # `add_dependencies`.
  add_custom_target(rebuild_${TARGET_NAME}
    DEPENDS dummy_rebulid_${TARGET_NAME}
    )

  # Add dummy code to support `make target_name` under Terminal Command
L
liaogang 已提交
327
  file(WRITE ${dummyfile} "const char * dummy = \"${dummyfile}\";")
L
liaogang 已提交
328 329 330 331 332
  if (go_library_SHARED OR go_library_shared)
    add_library(${TARGET_NAME} SHARED ${dummyfile})
  else()
    add_library(${TARGET_NAME} STATIC ${dummyfile})
  endif()
333 334 335
  if(go_library_DEPS)
    add_dependencies(${TARGET_NAME} ${go_library_DEPS})
  endif(go_library_DEPS)
L
liaogang 已提交
336

337 338 339 340 341 342
  # The "source file" of the library is `${dummyfile}` which never
  # change, so the target will never rebuild. Make the target depends
  # on the custom command that touches the library "source file", so
  # rebuild will always happen.
  add_dependencies(${TARGET_NAME} rebuild_${TARGET_NAME})

H
Helin Wang 已提交
343 344
  set(${TARGET_NAME}_LIB_PATH "${CMAKE_CURRENT_BINARY_DIR}/${${TARGET_NAME}_LIB_NAME}" CACHE STRING "output library path for target ${TARGET_NAME}")

L
liaogang 已提交
345
  file(GLOB GO_SOURCE RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "*.go")
346
  string(REPLACE "${PADDLE_GO_PATH}/" "" CMAKE_CURRENT_SOURCE_REL_DIR ${CMAKE_CURRENT_SOURCE_DIR})
347

L
liaogang 已提交
348
  add_custom_command(TARGET ${TARGET_NAME} POST_BUILD
H
Helin Wang 已提交
349
    COMMAND rm "${${TARGET_NAME}_LIB_PATH}"
350
    # Golang build source code
Y
yi.wu 已提交
351
    COMMAND GOPATH=${GOPATH} ${CMAKE_Go_COMPILER} build ${BUILD_MODE}
H
Helin Wang 已提交
352
    -o "${${TARGET_NAME}_LIB_PATH}"
W
wuyi05 已提交
353 354
    "./${CMAKE_CURRENT_SOURCE_REL_DIR}/${GO_SOURCE}"
    # must run under GOPATH
355
    WORKING_DIRECTORY "${PADDLE_IN_GOPATH}/go")
356
  add_dependencies(${TARGET_NAME} go_vendor)
L
liaogang 已提交
357 358 359 360 361 362 363
endfunction(go_library)

function(go_binary TARGET_NAME)
  set(options OPTIONAL)
  set(oneValueArgs "")
  set(multiValueArgs SRCS DEPS)
  cmake_parse_arguments(go_binary "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
364
  string(REPLACE "${PADDLE_GO_PATH}/" "" CMAKE_CURRENT_SOURCE_REL_DIR ${CMAKE_CURRENT_SOURCE_DIR})
W
wuyi05 已提交
365

L
liaogang 已提交
366
  add_custom_command(OUTPUT ${TARGET_NAME}_timestamp
367
    COMMAND env GOPATH=${GOPATH} ${CMAKE_Go_COMPILER} build
L
liaogang 已提交
368
    -o "${CMAKE_CURRENT_BINARY_DIR}/${TARGET_NAME}"
369
    "./${CMAKE_CURRENT_SOURCE_REL_DIR}/${go_binary_SRCS}"
G
gongweibao 已提交
370
    WORKING_DIRECTORY "${PADDLE_IN_GOPATH}/go")
371
  add_custom_target(${TARGET_NAME} ALL DEPENDS go_vendor ${TARGET_NAME}_timestamp ${go_binary_DEPS})
L
liaogang 已提交
372 373 374 375 376 377
  install(PROGRAMS ${CMAKE_CURRENT_BINARY_DIR}/${TARGET_NAME} DESTINATION bin)
endfunction(go_binary)

function(go_test TARGET_NAME)
  set(options OPTIONAL)
  set(oneValueArgs "")
378
  set(multiValueArgs DEPS)
L
liaogang 已提交
379
  cmake_parse_arguments(go_test "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
380 381 382
  string(REPLACE "${PADDLE_GO_PATH}" "" CMAKE_CURRENT_SOURCE_REL_DIR ${CMAKE_CURRENT_SOURCE_DIR})
  add_custom_target(${TARGET_NAME} ALL DEPENDS go_vendor ${go_test_DEPS})
  add_custom_command(TARGET ${TARGET_NAME} POST_BUILD
383
    COMMAND env GOPATH=${GOPATH} ${CMAKE_Go_COMPILER} test -race
L
liaogang 已提交
384
    -c -o "${CMAKE_CURRENT_BINARY_DIR}/${TARGET_NAME}"
385 386 387 388
    ".${CMAKE_CURRENT_SOURCE_REL_DIR}"
    WORKING_DIRECTORY "${PADDLE_IN_GOPATH}/go")
  add_test(NAME ${TARGET_NAME}
    COMMAND ${CMAKE_CURRENT_BINARY_DIR}/${TARGET_NAME}
389
    WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR})
L
liaogang 已提交
390
endfunction(go_test)
Y
Yu Yang 已提交
391

Y
Yu Yang 已提交
392
function(proto_library TARGET_NAME)
Y
Yu Yang 已提交
393
  set(oneValueArgs "")
Y
Yu Yang 已提交
394
  set(multiValueArgs SRCS DEPS)
Y
Yu Yang 已提交
395
  cmake_parse_arguments(proto_library "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
Y
Yu Yang 已提交
396 397
  set(proto_srcs)
  set(proto_hdrs)
Y
Yu Yang 已提交
398
  protobuf_generate_cpp(proto_srcs proto_hdrs ${proto_library_SRCS})
Y
Yu Yang 已提交
399
  cc_library(${TARGET_NAME} SRCS ${proto_srcs} DEPS ${proto_library_DEPS} protobuf)
Y
Yu Yang 已提交
400
endfunction()
401 402 403 404 405 406 407 408

function(py_proto_compile TARGET_NAME)
  set(oneValueArgs "")
  set(multiValueArgs SRCS)
  cmake_parse_arguments(py_proto_compile "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
  set(py_srcs)
  protobuf_generate_python(py_srcs ${py_proto_compile_SRCS})
  add_custom_target(${TARGET_NAME} ALL DEPENDS ${py_srcs})
409
endfunction()
L
liaogang 已提交
410 411 412 413 414 415 416 417

function(py_test TARGET_NAME)
  if(WITH_TESTING)
    set(options STATIC static SHARED shared)
    set(oneValueArgs "")
    set(multiValueArgs SRCS DEPS)
    cmake_parse_arguments(py_test "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})  
    add_test(NAME ${TARGET_NAME}
Q
qiaolongfei 已提交
418
             COMMAND env PYTHONPATH=${PADDLE_PYTHON_BUILD_DIR}/lib-python
L
liaogang 已提交
419 420 421 422
             python2 ${py_test_SRCS}
             WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR})
  endif()
endfunction()