61 lines
1.8 KiB
61 lines
1.8 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)
|
|
list(APPEND cpp_files GemmFunctor.cpp)
|
|
if(USE_EIGEN_FOR_BLAS)
|
|
list(APPEND cpp_files EigenGemm.cpp)
|
|
endif(USE_EIGEN_FOR_BLAS)
|
|
|
|
if(WITH_GPU)
|
|
file(GLOB cu_files . *OpGpu.cu)
|
|
cuda_compile(cu_objs ${cu_files})
|
|
endif()
|
|
|
|
if(USE_NNPACK)
|
|
list(APPEND cpp_files nnpack/NNPACKConvOp.cpp)
|
|
if(WITH_TESTING)
|
|
add_unittest(NNPACKConvOpTest nnpack/NNPACKConvOpTest.cpp)
|
|
endif()
|
|
endif()
|
|
|
|
list(APPEND cpp_files neon/NeonDepthwiseConv.cpp)
|
|
|
|
add_library(paddle_function STATIC ${cpp_files} ${cu_objs})
|
|
add_dependencies(paddle_function ${external_project_dependencies})
|
|
add_dependencies(paddle_function paddle_proto)
|
|
|
|
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)
|
|
add_simple_unittest(BlockExpandOpTest)
|
|
add_simple_unittest(CropOpTest)
|
|
add_simple_unittest(SwitchOpTest)
|
|
add_simple_unittest(ScaleSubRegionOpTest)
|
|
endif()
|
|
|
|
add_simple_unittest(Im2ColTest)
|
|
add_simple_unittest(GemmConvOpTest)
|
|
add_simple_unittest(DepthwiseConvOpTest)
|
|
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()
|