CMakeLists.txt 1.5 KB
Newer Older
H
hedaoyuan 已提交
1 2
file(GLOB h_files . *Op.h)
file(GLOB cpp_files . *Op.cpp)
3 4 5

list(APPEND h_files Function.h)
list(APPEND cpp_files Function.cpp)
6
list(APPEND cpp_files BufferArg.cpp)
7 8

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

13 14
if(USE_NNPACK)
  list(APPEND cpp_files nnpack/NNPACKConvOp.cpp)
H
hedaoyuan 已提交
15 16 17
  if(WITH_TESTING)
    add_unittest(NNPACKConvOpTest nnpack/NNPACKConvOpTest.cpp)
  endif()
18 19
endif()

20
add_library(paddle_function STATIC ${cpp_files} ${cu_objs})
21
add_dependencies(paddle_function ${external_project_dependencies})
Y
Yu Yang 已提交
22
add_dependencies(paddle_function paddle_proto)
23

H
hedaoyuan 已提交
24
if(WITH_TESTING)
25
if(WITH_GPU)
26
    # TODO:
H
hedaoyuan 已提交
27
    # file(GLOB test_files . *OpTest.cpp)
28
    # add_executable(${test_bin} EXCLUDE_FROM_ALL ${test_files})
H
hedaoyuan 已提交
29
    add_simple_unittest(CrossMapNormalOpTest)
30 31 32
    add_simple_unittest(TensorShapeTest)
    add_simple_unittest(TensorTypeTest)
    add_simple_unittest(BufferArgTest)
H
hedaoyuan 已提交
33
    add_simple_unittest(FunctionTest)
34
    add_simple_unittest(ContextProjectionOpTest)
D
dangqingqing 已提交
35
    add_simple_unittest(PadOpTest)
36
    add_simple_unittest(MulOpTest)
37
    add_simple_unittest(CosSimOpTest)
D
dangqingqing 已提交
38
    add_simple_unittest(RowConvOpTest)
H
hedaoyuan 已提交
39
    add_simple_unittest(BlockExpandOpTest)
W
wanghaoshuang 已提交
40
    add_simple_unittest(CropOpTest)
41
endif()
42 43

add_simple_unittest(ConvOpTest)
H
hedaoyuan 已提交
44
add_simple_unittest(Im2ColTest)
H
hedaoyuan 已提交
45
endif()
46 47 48 49 50 51

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()