提交 558e8692 编写于 作者: H hedaoyuan

add CMakeLists

上级 22a5e478
...@@ -96,6 +96,7 @@ function(link_paddle_exe TARGET_NAME) ...@@ -96,6 +96,7 @@ function(link_paddle_exe TARGET_NAME)
target_circle_link_libraries(${TARGET_NAME} target_circle_link_libraries(${TARGET_NAME}
ARCHIVE_START ARCHIVE_START
paddle_gserver paddle_gserver
paddle_function
${METRIC_LIBS} ${METRIC_LIBS}
ARCHIVE_END ARCHIVE_END
paddle_pserver paddle_pserver
......
add_subdirectory(cuda) add_subdirectory(cuda)
add_subdirectory(function)
add_subdirectory(utils) add_subdirectory(utils)
add_subdirectory(math) add_subdirectory(math)
add_subdirectory(parameter) add_subdirectory(parameter)
......
file(GLOB FUNCTION_HEADERS . *.h)
if(NOT WITH_GPU)
file(GLOB FUNCTION_SOURCES . *.cpp)
add_library(paddle_function STATIC ${FUNCTION_SOURCES})
else()
file(GLOB FUNCTION_SOURCES . *.cpp *.cu)
cuda_add_library(paddle_function ${FUNCTION_SOURCES})
endif()
add_style_check_target(paddle_function ${FUNCTION_SOURCES})
add_style_check_target(paddle_function ${FUNCTION_HEADERS})
...@@ -215,8 +215,10 @@ private: ...@@ -215,8 +215,10 @@ private:
}; };
REGISTER_TYPED_FUNC(CrossMapNormal, CPU, CrossMapNormalFunc); REGISTER_TYPED_FUNC(CrossMapNormal, CPU, CrossMapNormalFunc);
REGISTER_TYPED_FUNC(CrossMapNormal, GPU, CrossMapNormalFunc);
REGISTER_TYPED_FUNC(CrossMapNormalGrad, CPU, CrossMapNormalGradFunc); REGISTER_TYPED_FUNC(CrossMapNormalGrad, CPU, CrossMapNormalGradFunc);
#ifndef PADDLE_ONLY_CPU
REGISTER_TYPED_FUNC(CrossMapNormal, GPU, CrossMapNormalFunc);
REGISTER_TYPED_FUNC(CrossMapNormalGrad, GPU, CrossMapNormalGradFunc); REGISTER_TYPED_FUNC(CrossMapNormalGrad, GPU, CrossMapNormalGradFunc);
#endif
} // namespace paddle } // namespace paddle
...@@ -27,16 +27,12 @@ if(NOT WITH_GPU) ...@@ -27,16 +27,12 @@ if(NOT WITH_GPU)
list(REMOVE_ITEM GSERVER_HEADER list(REMOVE_ITEM GSERVER_HEADER
layers/CudnnConvLayer.h layers/CudnnConvLayer.h
layers/CudnnPoolLayer.h layers/CudnnPoolLayer.h
layers/CudnnBatchNormLayer.h layers/CudnnBatchNormLayer.h)
layers/NormProjectionLayer.h
layers/NormLayer.h)
list(REMOVE_ITEM GSERVER_SOURCES list(REMOVE_ITEM GSERVER_SOURCES
layers/CudnnConvLayer.cpp layers/CudnnConvLayer.cpp
layers/CudnnPoolLayer.cpp layers/CudnnPoolLayer.cpp
layers/CudnnBatchNormLayer.cpp layers/CudnnBatchNormLayer.cpp)
layers/NormProjectionLayer.cpp
layers/NormLayer.cpp)
compile_cu_as_cpp(layers/LstmCompute.cu) compile_cu_as_cpp(layers/LstmCompute.cu)
compile_cu_as_cpp(layers/GruCompute.cu) compile_cu_as_cpp(layers/GruCompute.cu)
endif() endif()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册