@ -66,23 +66,34 @@ if(WITH_PYTHON)
get_property ( os_dependency_modules GLOBAL PROPERTY OS_DEPENDENCY_MODULES )
target_link_libraries ( op_function_generator ${ os_dependency_modules } )
set ( impl_file ${ CMAKE_SOURCE_DIR } /paddle/fluid/pybind/op_function_impl.h )
set ( tmp_impl_file ${ impl_file } .tmp )
if ( WIN32 )
add_custom_target ( op_function_cmd
C O M M A N D " $ { C M A K E _ B I N A R Y _ D I R } / p a d d l e / f l u i d / p y b i n d / $ { C M A K E _ B U I L D _ T Y P E } / o p _ f u n c t i o n _ g e n e r a t o r "
" $ { C M A K E _ S O U R C E _ D I R } / p a d d l e / f l u i d / p y b i n d / o p _ f u n c t i o n _ i m p l . h " )
add_dependencies ( op_function_cmd op_function_generator )
add_custom_command ( TARGET op_function_generator
P O S T _ B U I L D
C O M M A N D " $ { C M A K E _ B I N A R Y _ D I R } / p a d d l e / f l u i d / p y b i n d / $ { C M A K E _ B U I L D _ T Y P E } / o p _ f u n c t i o n _ g e n e r a t o r "
" $ { t m p _ i m p l _ f i l e } "
C O M M A N D $ { C M A K E _ C O M M A N D } - E c o p y _ i f _ d i f f e r e n t $ { t m p _ i m p l _ f i l e } $ { i m p l _ f i l e }
C O M M E N T " c o p y _ i f _ d i f f e r e n t $ { i m p l _ f i l e } "
V E R B A T I M
)
if ( ${ CBLAS_PROVIDER } STREQUAL MKLML )
add_custom_command ( TARGET op_function_generator POST_BUILD
add_custom_command ( TARGET op_function_generator
P R E _ B U I L D
C O M M A N D $ { C M A K E _ C O M M A N D } - E c o p y $ { M K L M L _ S H A R E D _ L I B } $ { C M A K E _ C U R R E N T _ B I N A R Y _ D I R } / $ { C M A K E _ B U I L D _ T Y P E }
C O M M A N D $ { C M A K E _ C O M M A N D } - E c o p y $ { M K L M L _ S H A R E D _ I O M P _ L I B } $ { C M A K E _ C U R R E N T _ B I N A R Y _ D I R } / $ { C M A K E _ B U I L D _ T Y P E }
)
else ( ${ CBLAS_PROVIDER } STREQUAL EXTERN_OPENBLAS )
add_custom_command ( TARGET op_function_generator POST_BUILD
add_custom_command ( TARGET op_function_generator
P R E _ B U I L D
C O M M A N D $ { C M A K E _ C O M M A N D } - E c o p y $ { O P E N B L A S _ S H A R E D _ L I B } $ { C M A K E _ C U R R E N T _ B I N A R Y _ D I R } / $ { C M A K E _ B U I L D _ T Y P E }
)
endif ( )
if ( WITH_MKLDNN )
add_custom_command ( TARGET op_function_generator POST_BUILD
add_custom_command ( TARGET op_function_generator
P R E _ B U I L D
C O M M A N D $ { C M A K E _ C O M M A N D } - E c o p y $ { M K L D N N _ S H A R E D _ L I B } $ { C M A K E _ C U R R E N T _ B I N A R Y _ D I R } / $ { C M A K E _ B U I L D _ T Y P E }
)
endif ( )
@ -91,19 +102,25 @@ if(WITH_PYTHON)
# c o p y t h e s e * . s o t o c u r r e n t d i r e c t o r y a n d a p p e n d c u r r e n t d i r e c t o r y t o
# L D _ L I B R A R Y _ P A T H . T h i s i s d i f f e r e n t w i t h W i n d o w s p l a t f o r m m , w h i c h s e a r c h
# * . d l l i n c u r r e n t d i r e c t o r y a u t o m a t i c a l l y .
add_custom_target ( op_function_cmd
C O M M A N D $ { C M A K E _ C O M M A N D } - E e n v " L D _ L I B R A R Y _ P A T H = $ E N V { L D _ L I B R A R Y _ P A T H } : . "
add_custom_command ( TARGET op_function_generator
P O S T _ B U I L D
C O M M A N D $ { C M A K E _ C O M M A N D } - E e n v " L D _ L I B R A R Y _ P A T H = $ E N V { L D _ L I B R A R Y _ P A T H } : . "
" $ { C M A K E _ C U R R E N T _ B I N A R Y _ D I R } / o p _ f u n c t i o n _ g e n e r a t o r "
" $ { C M A K E _ S O U R C E _ D I R } / p a d d l e / f l u i d / p y b i n d / o p _ f u n c t i o n _ i m p l . h " )
add_dependencies ( op_function_cmd op_function_generator )
" $ { t m p _ i m p l _ f i l e } "
C O M M A N D $ { C M A K E _ C O M M A N D } - E c o p y _ i f _ d i f f e r e n t $ { t m p _ i m p l _ f i l e } $ { i m p l _ f i l e }
C O M M E N T " c o p y _ i f _ d i f f e r e n t $ { i m p l _ f i l e } "
V E R B A T I M
)
if ( WITH_MKL )
add_custom_command ( TARGET op_function_generator POST_BUILD
add_custom_command ( TARGET op_function_generator
P R E _ B U I L D
C O M M A N D $ { C M A K E _ C O M M A N D } - E c o p y $ { M K L M L _ S H A R E D _ L I B } $ { C M A K E _ C U R R E N T _ B I N A R Y _ D I R }
C O M M A N D $ { C M A K E _ C O M M A N D } - E c o p y $ { M K L M L _ S H A R E D _ I O M P _ L I B } $ { C M A K E _ C U R R E N T _ B I N A R Y _ D I R }
)
endif ( WITH_MKL )
if ( WITH_MKLDNN )
add_custom_command ( TARGET op_function_generator POST_BUILD
add_custom_command ( TARGET op_function_generator
P R E _ B U I L D
C O M M A N D $ { C M A K E _ C O M M A N D } - E c o p y $ { M K L D N N _ S H A R E D _ L I B } $ { C M A K E _ C U R R E N T _ B I N A R Y _ D I R }
)
endif ( WITH_MKLDNN )
@ -126,5 +143,5 @@ if(WITH_PYTHON)
get_property ( os_dependency_modules GLOBAL PROPERTY OS_DEPENDENCY_MODULES )
target_link_libraries ( paddle_pybind ${ os_dependency_modules } )
add_dependencies ( paddle_pybind op_function_ cmd )
add_dependencies ( paddle_pybind op_function_ generator )
endif ( WITH_PYTHON )
Shibo Tao
5 years ago
committed by
GitHub