42 lines
1.2 KiB
42 lines
1.2 KiB
file(GLOB h_files . *Op.h)
|
|
file(GLOB cpp_files . *Op.cpp)
|
|
|
|
list(APPEND h_files Function.h)
|
|
list(APPEND cpp_files Function.cpp)
|
|
list(APPEND cpp_files BufferArg.cpp)
|
|
|
|
if(WITH_GPU)
|
|
file(GLOB cu_files . *OpGpu.cu)
|
|
cuda_compile(cu_objs ${cu_files})
|
|
endif()
|
|
|
|
add_library(paddle_function STATIC ${cpp_files} ${cu_objs})
|
|
add_dependencies(paddle_function ${external_project_dependencies})
|
|
add_dependencies(paddle_function gen_proto_cpp)
|
|
|
|
if(WITH_TESTING)
|
|
if(WITH_GPU)
|
|
# TODO:
|
|
# file(GLOB test_files . *OpTest.cpp)
|
|
# add_executable(${test_bin} EXCLUDE_FROM_ALL ${test_files})
|
|
add_simple_unittest(CrossMapNormalOpTest)
|
|
add_simple_unittest(TensorShapeTest)
|
|
add_simple_unittest(TensorTypeTest)
|
|
add_simple_unittest(BufferArgTest)
|
|
add_simple_unittest(FunctionTest)
|
|
add_simple_unittest(ContextProjectionOpTest)
|
|
add_simple_unittest(PadOpTest)
|
|
add_simple_unittest(MulOpTest)
|
|
add_simple_unittest(CosSimOpTest)
|
|
add_simple_unittest(RowConvOpTest)
|
|
endif()
|
|
|
|
add_simple_unittest(ConvOpTest)
|
|
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()
|