Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Crayon鑫
Paddle
提交
aab4d6e4
P
Paddle
项目概览
Crayon鑫
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
aab4d6e4
编写于
8月 09, 2021
作者:
zhouweiwei2014
提交者:
GitHub
8月 09, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Increase the speed of incremental compilation (#34616)
上级
4c1ba73f
变更
7
隐藏空白更改
内联
并排
Showing
7 changed file
with
81 addition
and
88 deletion
+81
-88
CMakeLists.txt
CMakeLists.txt
+1
-1
cmake/external/gflags.cmake
cmake/external/gflags.cmake
+1
-0
cmake/external/glog.cmake
cmake/external/glog.cmake
+1
-0
cmake/external/mkldnn.cmake
cmake/external/mkldnn.cmake
+25
-30
cmake/external/protobuf.cmake
cmake/external/protobuf.cmake
+10
-10
cmake/external/pybind11.cmake
cmake/external/pybind11.cmake
+1
-0
paddle/fluid/pybind/CMakeLists.txt
paddle/fluid/pybind/CMakeLists.txt
+42
-47
未找到文件。
CMakeLists.txt
浏览文件 @
aab4d6e4
...
@@ -130,7 +130,7 @@ if(WIN32)
...
@@ -130,7 +130,7 @@ if(WIN32)
# NOTE(zhouwei25): GPU compile have too high memory utilization when parallel compiling,
# NOTE(zhouwei25): GPU compile have too high memory utilization when parallel compiling,
# For Visual Studio generators, /MP should be added.
# For Visual Studio generators, /MP should be added.
# For other generators like Ninja, it is not need to add /MP.
# For other generators like Ninja, it is not need to add /MP.
if
(
"
${
CMAKE_GENERATOR
}
"
STREQUAL
"Visual Studio"
AND NOT WITH_GPU
)
if
(
CMAKE_GENERATOR MATCHES
"Visual Studio"
AND NOT WITH_GPU
)
math
(
EXPR PROCESS_MAX
"
${
CPU_CORES
}
* 2 / 3"
)
math
(
EXPR PROCESS_MAX
"
${
CPU_CORES
}
* 2 / 3"
)
set
(
${
flag_var
}
"
${${
flag_var
}}
/MP
${
PROCESS_MAX
}
"
)
set
(
${
flag_var
}
"
${${
flag_var
}}
/MP
${
PROCESS_MAX
}
"
)
endif
()
endif
()
...
...
cmake/external/gflags.cmake
浏览文件 @
aab4d6e4
...
@@ -41,6 +41,7 @@ ExternalProject_Add(
...
@@ -41,6 +41,7 @@ ExternalProject_Add(
${
SHALLOW_CLONE
}
${
SHALLOW_CLONE
}
"
${
GFLAGS_DOWNLOAD_CMD
}
"
"
${
GFLAGS_DOWNLOAD_CMD
}
"
PREFIX
${
GFLAGS_PREFIX_DIR
}
PREFIX
${
GFLAGS_PREFIX_DIR
}
UPDATE_COMMAND
""
SOURCE_DIR
${
GFLAGS_SOURCE_DIR
}
SOURCE_DIR
${
GFLAGS_SOURCE_DIR
}
BUILD_COMMAND
${
BUILD_COMMAND
}
BUILD_COMMAND
${
BUILD_COMMAND
}
INSTALL_COMMAND
${
INSTALL_COMMAND
}
INSTALL_COMMAND
${
INSTALL_COMMAND
}
...
...
cmake/external/glog.cmake
浏览文件 @
aab4d6e4
...
@@ -45,6 +45,7 @@ ExternalProject_Add(
...
@@ -45,6 +45,7 @@ ExternalProject_Add(
DEPENDS gflags
DEPENDS gflags
PREFIX
${
GLOG_PREFIX_DIR
}
PREFIX
${
GLOG_PREFIX_DIR
}
SOURCE_DIR
${
GLOG_SOURCE_DIR
}
SOURCE_DIR
${
GLOG_SOURCE_DIR
}
UPDATE_COMMAND
""
CMAKE_ARGS -DCMAKE_CXX_COMPILER=
${
CMAKE_CXX_COMPILER
}
CMAKE_ARGS -DCMAKE_CXX_COMPILER=
${
CMAKE_CXX_COMPILER
}
-DCMAKE_C_COMPILER=
${
CMAKE_C_COMPILER
}
-DCMAKE_C_COMPILER=
${
CMAKE_C_COMPILER
}
-DCMAKE_CXX_FLAGS=
${
GLOG_CMAKE_CXX_FLAGS
}
-DCMAKE_CXX_FLAGS=
${
GLOG_CMAKE_CXX_FLAGS
}
...
...
cmake/external/mkldnn.cmake
浏览文件 @
aab4d6e4
...
@@ -79,22 +79,10 @@ ExternalProject_Add(
...
@@ -79,22 +79,10 @@ ExternalProject_Add(
-DCMAKE_CXX_FLAGS=
${
MKLDNN_CXXFLAG
}
-DCMAKE_CXX_FLAGS=
${
MKLDNN_CXXFLAG
}
-DDNNL_BUILD_TESTS=OFF -DDNNL_BUILD_EXAMPLES=OFF
-DDNNL_BUILD_TESTS=OFF -DDNNL_BUILD_EXAMPLES=OFF
CMAKE_CACHE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=
${
MKLDNN_INSTALL_DIR
}
CMAKE_CACHE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=
${
MKLDNN_INSTALL_DIR
}
BUILD_BYPRODUCTS
${
MKLDNN_LIB
}
)
)
ADD_LIBRARY
(
shared_mkldnn SHARED IMPORTED GLOBAL
)
SET_PROPERTY
(
TARGET shared_mkldnn PROPERTY IMPORTED_LOCATION
${
MKLDNN_LIB
}
)
ADD_DEPENDENCIES
(
shared_mkldnn
${
MKLDNN_PROJECT
}
)
MESSAGE
(
STATUS
"MKLDNN library:
${
MKLDNN_LIB
}
"
)
MESSAGE
(
STATUS
"MKLDNN library:
${
MKLDNN_LIB
}
"
)
add_definitions
(
-DPADDLE_WITH_MKLDNN
)
add_definitions
(
-DPADDLE_WITH_MKLDNN
)
# generate a static dummy target to track mkldnn dependencies
# for cc_library(xxx SRCS xxx.c DEPS mkldnn)
generate_dummy_static_lib
(
LIB_NAME
"mkldnn"
GENERATOR
"mkldnn.cmake"
)
TARGET_LINK_LIBRARIES
(
mkldnn
${
MKLDNN_LIB
}
${
MKLML_IOMP_LIB
}
)
ADD_DEPENDENCIES
(
mkldnn
${
MKLDNN_PROJECT
}
)
# copy the real so.0 lib to install dir
# copy the real so.0 lib to install dir
# it can be directly contained in wheel or capi
# it can be directly contained in wheel or capi
if
(
WIN32
)
if
(
WIN32
)
...
@@ -102,26 +90,33 @@ if(WIN32)
...
@@ -102,26 +90,33 @@ if(WIN32)
file
(
TO_NATIVE_PATH
${
MKLDNN_INSTALL_DIR
}
NATIVE_MKLDNN_INSTALL_DIR
)
file
(
TO_NATIVE_PATH
${
MKLDNN_INSTALL_DIR
}
NATIVE_MKLDNN_INSTALL_DIR
)
file
(
TO_NATIVE_PATH
${
MKLDNN_SHARED_LIB
}
NATIVE_MKLDNN_SHARED_LIB
)
file
(
TO_NATIVE_PATH
${
MKLDNN_SHARED_LIB
}
NATIVE_MKLDNN_SHARED_LIB
)
ADD_CUSTOM_COMMAND
(
TARGET
${
MKLDNN_PROJECT
}
POST_BUILD
COMMAND
(
copy
${
NATIVE_MKLDNN_INSTALL_DIR
}
\\bin\\dnnl.dll
${
NATIVE_MKLDNN_SHARED_LIB
}
/Y
))
ADD_CUSTOM_COMMAND
(
OUTPUT
${
MKLDNN_LIB
}
add_custom_command
(
TARGET
${
MKLDNN_PROJECT
}
POST_BUILD VERBATIM
COMMAND
(
copy
${
NATIVE_MKLDNN_INSTALL_DIR
}
\\bin\\dnnl.dll
${
NATIVE_MKLDNN_SHARED_LIB
}
/Y
)
COMMAND dumpbin /exports
${
MKLDNN_INSTALL_DIR
}
/bin/mkldnn.dll >
${
MKLDNN_INSTALL_DIR
}
/bin/exports.txt
)
COMMAND dumpbin /exports
${
MKLDNN_INSTALL_DIR
}
/bin/mkldnn.dll >
${
MKLDNN_INSTALL_DIR
}
/bin/exports.txt
add_custom_command
(
TARGET
${
MKLDNN_PROJECT
}
POST_BUILD VERBATIM
COMMAND echo LIBRARY mkldnn >
${
MKLDNN_INSTALL_DIR
}
/bin/mkldnn.def
COMMAND echo
LIBRARY mkldnn >
${
MKLDNN_INSTALL_DIR
}
/bin/mkldnn.def
)
COMMAND echo
EXPORTS >>
${
MKLDNN_INSTALL_DIR
}
/bin/mkldnn.def
add_custom_command
(
TARGET
${
MKLDNN_PROJECT
}
POST_BUILD VERBATIM
COMMAND echo off &&
(
for /f
"skip=19 tokens=4"
%A
in
(
${
MKLDNN_INSTALL_DIR
}
/bin/exports.txt
)
do echo %A >>
${
MKLDNN_INSTALL_DIR
}
/bin/mkldnn.def
)
&& echo on
COMMAND
echo EXPORTS >>
${
MKLDNN_INSTALL_DIR
}
/bin/mkldnn.def
)
COMMAND
lib /def:
${
MKLDNN_INSTALL_DIR
}
/bin/mkldnn.def /out:
${
MKLDNN_LIB
}
/machine:x64
add_custom_command
(
TARGET
${
MKLDNN_PROJECT
}
POST_BUILD VERBATIM
COMMENT
"Generate mkldnn.lib manually--->"
COMMAND echo off &&
(
for /f
"skip=19 tokens=4"
%A
in
(
${
MKLDNN_INSTALL_DIR
}
/bin/exports.txt
)
do echo %A >>
${
MKLDNN_INSTALL_DIR
}
/bin/mkldnn.def
)
&& echo on
)
DEPENDS
${
MKLDNN_PROJECT
}
add_custom_command
(
TARGET
${
MKLDNN_PROJECT
}
POST_BUILD VERBATIM
VERBATIM
)
COMMAND lib /def:
${
MKLDNN_INSTALL_DIR
}
/bin/mkldnn.def /out:
${
MKLDNN_INSTALL_DIR
}
/bin/mkldnn.lib /machine:x64
)
ADD_CUSTOM_TARGET
(
mkldnn_cmd ALL DEPENDS
${
MKLDNN_LIB
}
)
else
(
WIN32
)
else
(
WIN32
)
SET
(
MKLDNN_SHARED_LIB
${
MKLDNN_INSTALL_DIR
}
/libmkldnn.so.0
)
SET
(
MKLDNN_SHARED_LIB
${
MKLDNN_INSTALL_DIR
}
/libmkldnn.so.0
)
SET
(
MKLDNN_SHARED_LIB_1
${
MKLDNN_INSTALL_DIR
}
/libdnnl.so.1
)
SET
(
MKLDNN_SHARED_LIB_1
${
MKLDNN_INSTALL_DIR
}
/libdnnl.so.1
)
SET
(
MKLDNN_SHARED_LIB_2
${
MKLDNN_INSTALL_DIR
}
/libdnnl.so.2
)
SET
(
MKLDNN_SHARED_LIB_2
${
MKLDNN_INSTALL_DIR
}
/libdnnl.so.2
)
ADD_CUSTOM_COMMAND
(
TARGET
${
MKLDNN_PROJECT
}
POST_BUILD
ADD_CUSTOM_COMMAND
(
OUTPUT
${
MKLDNN_SHARED_LIB_2
}
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
MKLDNN_LIB
}
${
MKLDNN_SHARED_LIB
}
)
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
MKLDNN_LIB
}
${
MKLDNN_SHARED_LIB
}
ADD_CUSTOM_COMMAND
(
TARGET
${
MKLDNN_PROJECT
}
POST_BUILD
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
MKLDNN_LIB
}
${
MKLDNN_SHARED_LIB_1
}
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
MKLDNN_LIB
}
${
MKLDNN_SHARED_LIB_1
}
)
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
MKLDNN_LIB
}
${
MKLDNN_SHARED_LIB_2
}
ADD_CUSTOM_COMMAND
(
TARGET
${
MKLDNN_PROJECT
}
POST_BUILD
DEPENDS
${
MKLDNN_PROJECT
}
)
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
MKLDNN_LIB
}
${
MKLDNN_SHARED_LIB_2
}
)
ADD_CUSTOM_TARGET
(
mkldnn_cmd ALL DEPENDS
${
MKLDNN_SHARED_LIB_2
}
)
endif
(
WIN32
)
endif
(
WIN32
)
# generate a static dummy target to track mkldnn dependencies
# for cc_library(xxx SRCS xxx.c DEPS mkldnn)
generate_dummy_static_lib
(
LIB_NAME
"mkldnn"
GENERATOR
"mkldnn.cmake"
)
TARGET_LINK_LIBRARIES
(
mkldnn
${
MKLDNN_LIB
}
${
MKLML_IOMP_LIB
}
)
ADD_DEPENDENCIES
(
mkldnn
${
MKLDNN_PROJECT
}
mkldnn_cmd
)
cmake/external/protobuf.cmake
浏览文件 @
aab4d6e4
...
@@ -198,16 +198,16 @@ FUNCTION(build_protobuf TARGET_NAME BUILD_FOR_HOST)
...
@@ -198,16 +198,16 @@ FUNCTION(build_protobuf TARGET_NAME BUILD_FOR_HOST)
"-Dprotobuf_MSVC_STATIC_RUNTIME=
${
MSVC_STATIC_CRT
}
"
)
"-Dprotobuf_MSVC_STATIC_RUNTIME=
${
MSVC_STATIC_CRT
}
"
)
ENDIF
()
ENDIF
()
if
(
WITH_ASCEND AND NOT WITH_ASCEND_CXX11
)
if
(
WITH_ASCEND AND NOT WITH_ASCEND_CXX11
)
SET
(
PROTOBUF_REPOSITORY https://gitee.com/tianjianhe/protobuf.git
)
SET
(
PROTOBUF_REPOSITORY https://gitee.com/tianjianhe/protobuf.git
)
SET
(
PROTOBUF_TAG v3.8.0
)
SET
(
PROTOBUF_TAG v3.8.0
)
elseif
(
WITH_ASCEND_CL AND NOT WITH_ASCEND_CXX11
)
elseif
(
WITH_ASCEND_CL AND NOT WITH_ASCEND_CXX11
)
SET
(
PROTOBUF_REPOSITORY https://gitee.com/tianjianhe/protobuf.git
)
SET
(
PROTOBUF_REPOSITORY https://gitee.com/tianjianhe/protobuf.git
)
SET
(
PROTOBUF_TAG v3.8.0
)
SET
(
PROTOBUF_TAG v3.8.0
)
else
()
else
()
SET
(
PROTOBUF_REPOSITORY
${
GIT_URL
}
/protocolbuffers/protobuf.git
)
SET
(
PROTOBUF_REPOSITORY
${
GIT_URL
}
/protocolbuffers/protobuf.git
)
SET
(
PROTOBUF_TAG 9f75c5aa851cd877fb0d93ccc31b8567a6706546
)
SET
(
PROTOBUF_TAG 9f75c5aa851cd877fb0d93ccc31b8567a6706546
)
endif
()
endif
()
cache_third_party
(
${
TARGET_NAME
}
cache_third_party
(
${
TARGET_NAME
}
REPOSITORY
${
PROTOBUF_REPOSITORY
}
REPOSITORY
${
PROTOBUF_REPOSITORY
}
...
...
cmake/external/pybind11.cmake
浏览文件 @
aab4d6e4
...
@@ -39,6 +39,7 @@ ExternalProject_Add(
...
@@ -39,6 +39,7 @@ ExternalProject_Add(
# to be modified without triggering incremental compilation, and the
# to be modified without triggering incremental compilation, and the
# third-party library version changes cannot be incorporated.
# third-party library version changes cannot be incorporated.
# reference: https://cmake.org/cmake/help/latest/module/ExternalProject.html
# reference: https://cmake.org/cmake/help/latest/module/ExternalProject.html
UPDATE_COMMAND
""
CONFIGURE_COMMAND
""
CONFIGURE_COMMAND
""
BUILD_COMMAND
""
BUILD_COMMAND
""
INSTALL_COMMAND
""
INSTALL_COMMAND
""
...
...
paddle/fluid/pybind/CMakeLists.txt
浏览文件 @
aab4d6e4
...
@@ -124,23 +124,20 @@ if(WITH_PYTHON)
...
@@ -124,23 +124,20 @@ if(WITH_PYTHON)
set
(
impl_file
${
CMAKE_SOURCE_DIR
}
/paddle/fluid/pybind/op_function_impl.h
)
set
(
impl_file
${
CMAKE_SOURCE_DIR
}
/paddle/fluid/pybind/op_function_impl.h
)
set
(
tmp_impl_file
${
impl_file
}
.tmp
)
set
(
tmp_impl_file
${
impl_file
}
.tmp
)
set
(
OP_IMPL_DEPS op_function_generator
)
if
(
WIN32
)
if
(
WIN32
)
if
(
"
${
CMAKE_GENERATOR
}
"
STREQUAL
"Ninja"
)
if
(
"
${
CMAKE_GENERATOR
}
"
STREQUAL
"Ninja"
)
set
(
op_
function_generator
_path
"
${
CMAKE_CURRENT_BINARY_DIR
}
"
)
set
(
op_
impl
_path
"
${
CMAKE_CURRENT_BINARY_DIR
}
"
)
else
()
else
()
set
(
op_
function_generator
_path
"
${
CMAKE_CURRENT_BINARY_DIR
}
/
${
CMAKE_BUILD_TYPE
}
"
)
set
(
op_
impl
_path
"
${
CMAKE_CURRENT_BINARY_DIR
}
/
${
CMAKE_BUILD_TYPE
}
"
)
endif
()
endif
()
file
(
TO_NATIVE_PATH
${
op_function_generator_path
}
op_function_generator_path
)
file
(
TO_NATIVE_PATH
${
impl_file
}
impl_file
)
file
(
TO_NATIVE_PATH
${
tmp_impl_file
}
tmp_impl_file
)
file
(
WRITE
${
CMAKE_BINARY_DIR
}
/paddle/fluid/pybind/op_function_generator_retry.bat
""
file
(
WRITE
${
CMAKE_BINARY_DIR
}
/paddle/fluid/pybind/op_function_generator_retry.bat
""
"set build_times=1
\n
"
"set build_times=1
\n
"
":retry
\n
"
":retry
\n
"
"ECHO op_function_generator run %build_times% time
\n
"
"ECHO op_function_generator run %build_times% time
\n
"
"if exist
${
tmp_impl_file
}
del
${
tmp_impl_file
}
\n
"
"taskkill /f /im op_function_generator.exe 2>NUL
\n
"
"taskkill /f /im op_function_generator.exe 2>NUL
\n
"
"
${
op_
function_generator_path
}
\\
op_function_generator.exe
${
tmp_impl_file
}
\n
"
"
${
op_
impl_path
}
/
op_function_generator.exe
${
tmp_impl_file
}
\n
"
"if %ERRORLEVEL% NEQ 0 (
\n
"
"if %ERRORLEVEL% NEQ 0 (
\n
"
" set /a build_times=%build_times%+1
\n
"
" set /a build_times=%build_times%+1
\n
"
" if %build_times% GEQ 10 (
\n
"
" if %build_times% GEQ 10 (
\n
"
...
@@ -151,63 +148,61 @@ if(WITH_PYTHON)
...
@@ -151,63 +148,61 @@ if(WITH_PYTHON)
")
\n
"
")
\n
"
"exit /b 0"
)
"exit /b 0"
)
add_custom_command
(
TARGET op_function_generator POST_BUILD
COMMAND
${
CMAKE_BINARY_DIR
}
/paddle/fluid/pybind/op_function_generator_retry.bat
COMMAND
${
CMAKE_COMMAND
}
-E copy_if_different
${
tmp_impl_file
}
${
impl_file
}
COMMENT
"copy_if_different
${
tmp_impl_file
}
to
${
impl_file
}
"
)
if
(
${
CBLAS_PROVIDER
}
STREQUAL MKLML
)
if
(
${
CBLAS_PROVIDER
}
STREQUAL MKLML
)
add_custom_command
(
TARGET op_function_generator
ADD_CUSTOM_COMMAND
(
OUTPUT
${
op_impl_path
}
/libiomp5md.dll
PRE_LINK
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
MKLML_SHARED_IOMP_LIB
}
${
op_impl_path
}
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
MKLML_SHARED_LIB
}
${
op_function_generator_path
}
DEPENDS mklml
)
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
MKLML_SHARED_IOMP_LIB
}
${
op_function_generator_path
}
list
(
APPEND OP_IMPL_DEPS
${
op_impl_path
}
/libiomp5md.dll
)
)
else
(
${
CBLAS_PROVIDER
}
STREQUAL EXTERN_OPENBLAS
)
else
(
${
CBLAS_PROVIDER
}
STREQUAL EXTERN_OPENBLAS
)
add_custom_command
(
TARGET op_function_generator
ADD_CUSTOM_COMMAND
(
OUTPUT
${
op_impl_path
}
/openblas.dll
PRE_LINK
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
OPENBLAS_SHARED_LIB
}
${
op_impl_path
}
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
OPENBLAS_SHARED_LIB
}
${
op_function_generator_path
}
DEPENDS extern_openblas
)
)
list
(
APPEND OP_IMPL_DEPS
${
op_impl_path
}
/openblas.dll
)
endif
()
endif
()
if
(
WITH_MKLDNN
)
if
(
WITH_MKLDNN
)
add_custom_command
(
TARGET op_function_generator
ADD_CUSTOM_COMMAND
(
OUTPUT
${
op_impl_path
}
/mkldnn.dll
PRE_LINK
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
MKLDNN_SHARED_LIB
}
${
op_impl_path
}
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
MKLDNN_SHARED_LIB
}
${
op_function_generator_path
}
DEPENDS mkldnn
)
)
list
(
APPEND OP_IMPL_DEPS
${
op_impl_path
}
/mkldnn.dll
)
endif
()
endif
()
add_custom_command
(
OUTPUT
${
impl_file
}
COMMAND
${
CMAKE_BINARY_DIR
}
/paddle/fluid/pybind/op_function_generator_retry.bat
COMMAND
${
CMAKE_COMMAND
}
-E copy_if_different
${
tmp_impl_file
}
${
impl_file
}
COMMENT
"copy_if_different
${
tmp_impl_file
}
to
${
impl_file
}
"
DEPENDS
${
OP_IMPL_DEPS
}
)
else
(
WIN32
)
else
(
WIN32
)
# If there are no *.so in /usr/lib or LD_LIBRARY_PATH,
# If there are no *.so in /usr/lib or LD_LIBRARY_PATH,
# copy these *.so to current directory and append current directory to
# copy these *.so to current directory and append current directory to
# LD_LIBRARY_PATH. This is different with Windows platformm, which search
# LD_LIBRARY_PATH. This is different with Windows platformm, which search
# *.dll in current directory automatically.
# *.dll in current directory automatically.
add_custom_command
(
TARGET op_function_generator
if
(
WITH_MKLML
)
POST_BUILD
ADD_CUSTOM_COMMAND
(
OUTPUT
${
CMAKE_CURRENT_BINARY_DIR
}
/libiomp5.so
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
MKLML_SHARED_IOMP_LIB
}
${
CMAKE_CURRENT_BINARY_DIR
}
DEPENDS mklml
)
list
(
APPEND OP_IMPL_DEPS
${
CMAKE_CURRENT_BINARY_DIR
}
/libiomp5.so
)
endif
()
if
(
WITH_MKLDNN
)
ADD_CUSTOM_COMMAND
(
OUTPUT
${
CMAKE_CURRENT_BINARY_DIR
}
/libdnnl.so.0
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
MKLDNN_SHARED_LIB
}
${
CMAKE_CURRENT_BINARY_DIR
}
DEPENDS mkldnn
)
list
(
APPEND OP_IMPL_DEPS
${
CMAKE_CURRENT_BINARY_DIR
}
/libdnnl.so.0
)
endif
()
add_custom_command
(
OUTPUT
${
impl_file
}
COMMAND
${
CMAKE_COMMAND
}
-E env
"LD_LIBRARY_PATH=$ENV{LD_LIBRARY_PATH}:."
COMMAND
${
CMAKE_COMMAND
}
-E env
"LD_LIBRARY_PATH=$ENV{LD_LIBRARY_PATH}:."
"
${
CMAKE_CURRENT_BINARY_DIR
}
/op_function_generator"
"
${
CMAKE_CURRENT_BINARY_DIR
}
/op_function_generator"
"
${
tmp_impl_file
}
"
"
${
tmp_impl_file
}
"
COMMAND
${
CMAKE_COMMAND
}
-E copy_if_different
${
tmp_impl_file
}
${
impl_file
}
COMMAND
${
CMAKE_COMMAND
}
-E copy_if_different
${
tmp_impl_file
}
${
impl_file
}
COMMENT
"copy_if_different
${
tmp_impl_file
}
to
${
impl_file
}
"
COMMENT
"copy_if_different
${
tmp_impl_file
}
to
${
impl_file
}
"
VERBATIM
DEPENDS
${
OP_IMPL_DEPS
}
)
VERBATIM
)
if
(
WITH_MKL
)
add_custom_command
(
TARGET op_function_generator
PRE_LINK
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
MKLML_SHARED_LIB
}
${
CMAKE_CURRENT_BINARY_DIR
}
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
MKLML_SHARED_IOMP_LIB
}
${
CMAKE_CURRENT_BINARY_DIR
}
)
endif
(
WITH_MKL
)
if
(
WITH_MKLDNN
)
add_custom_command
(
TARGET op_function_generator
PRE_LINK
COMMAND
${
CMAKE_COMMAND
}
-E copy
${
MKLDNN_SHARED_LIB
}
${
CMAKE_CURRENT_BINARY_DIR
}
)
endif
(
WITH_MKLDNN
)
endif
(
WIN32
)
endif
(
WIN32
)
add_custom_target
(
op_function_generator_cmd ALL DEPENDS
${
impl_file
}
)
cc_library
(
paddle_pybind SHARED
cc_library
(
paddle_pybind SHARED
SRCS
${
PYBIND_SRCS
}
SRCS
${
PYBIND_SRCS
}
DEPS
${
PYBIND_DEPS
}
DEPS
${
PYBIND_DEPS
}
${
GLOB_OP_LIB
}
${
GLOB_OPERATOR_DEPS
}
)
${
GLOB_OP_LIB
}
${
GLOB_OPERATOR_DEPS
}
)
if
(
NOT APPLE AND NOT WIN32
)
if
(
NOT APPLE AND NOT WIN32
)
target_link_libraries
(
paddle_pybind rt
)
target_link_libraries
(
paddle_pybind rt
)
endif
(
NOT APPLE AND NOT WIN32
)
endif
(
NOT APPLE AND NOT WIN32
)
...
@@ -218,5 +213,5 @@ if(WITH_PYTHON)
...
@@ -218,5 +213,5 @@ if(WITH_PYTHON)
get_property
(
os_dependency_modules GLOBAL PROPERTY OS_DEPENDENCY_MODULES
)
get_property
(
os_dependency_modules GLOBAL PROPERTY OS_DEPENDENCY_MODULES
)
target_link_libraries
(
paddle_pybind
${
os_dependency_modules
}
)
target_link_libraries
(
paddle_pybind
${
os_dependency_modules
}
)
add_dependencies
(
paddle_pybind op_function_generator
)
add_dependencies
(
paddle_pybind op_function_generator
_cmd
)
endif
(
WITH_PYTHON
)
endif
(
WITH_PYTHON
)
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录