CMakeLists.txt 714 字节
Newer Older
1 2 3 4 5 6 7 8 9
file(GLOB h_files . *_op.h)
file(GLOB cpp_files . *_op.cpp)

list(APPEND h_files Function.h)
list(APPEND cpp_files Function.cpp)

if(WITH_GPU)
    file(GLOB cu_files . *_op_gpu.cu)
    cuda_compile(cu_objs ${cu_files})
H
hedaoyuan 已提交
10 11
endif()

12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
add_library(paddle_function STATIC ${cpp_files} ${cu_objs})

add_library(paddle_test_main STATIC TestMain.cpp)

if(WITH_GPU)
    # TODO:
    # file(GLOB test_files . *_op_test.cpp)
    # add_executable(${test_bin} EXCLUDE_FROM_ALL ${test_files})
    add_simple_unittest(cross_map_normal_op_test)
endif()

add_style_check_target(paddle_function ${h_files})
add_style_check_target(paddle_function ${cpp_files})
if(WITH_GPU)
    add_style_check_target(paddle_function ${cu_files})
endif()