forked from huawei/mindspore2022
25 lines
1.1 KiB
CMake
25 lines
1.1 KiB
CMake
file(GLOB_RECURSE _DISTRIBUTED_SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "*.cc")
|
|
|
|
if(NOT ENABLE_CPU OR WIN32)
|
|
list(REMOVE_ITEM _DISTRIBUTED_SRC_FILES "cluster/cluster_context.cc")
|
|
list(REMOVE_ITEM _DISTRIBUTED_SRC_FILES "cluster/actor_route_table_proxy.cc")
|
|
else()
|
|
list(REMOVE_ITEM _DISTRIBUTED_SRC_FILES "cluster/dummy_cluster_context.cc")
|
|
endif()
|
|
|
|
if(NOT ENABLE_CPU OR WIN32 OR APPLE)
|
|
set(EXCLUDE_DIR "rpc/")
|
|
foreach(RPC_FILE ${_DISTRIBUTED_SRC_FILES})
|
|
string(FIND ${RPC_FILE} ${EXCLUDE_DIR} FOUND)
|
|
if(${FOUND} EQUAL 0)
|
|
list(REMOVE_ITEM _DISTRIBUTED_SRC_FILES ${RPC_FILE})
|
|
endif()
|
|
endforeach()
|
|
endif()
|
|
|
|
set_property(SOURCE ${_DISTRIBUTED_SRC_FILES} PROPERTY COMPILE_DEFINITIONS
|
|
SUBMODULE_ID=mindspore::SubModuleId::SM_DISTRIBUTED)
|
|
add_library(_mindspore_distributed_obj OBJECT ${_DISTRIBUTED_SRC_FILES})
|
|
target_link_libraries(_mindspore_distributed_obj PRIVATE mindspore::protobuf mindspore::json mindspore::pybind11_module
|
|
$<$<BOOL:${USE_GLOG}>:mindspore::glog>)
|