zzy34407230
/
mindspore2022
Not watched
1
0
Fork 0
Code
Releases 22 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 062dcaaaa1
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '062dcaaaa1'
${ noResults }
mindspore2022/mindspore/ccsrc/ps/CMakeLists.txt

58 lines
3.1 kB
Raw Blame History 

  1. file(GLOB_RECURSE _PS_SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "*.cc")

  2. 

  3. if(NOT ENABLE_CPU OR WIN32)

  4.     list(REMOVE_ITEM _PS_SRC_FILES "optimizer_info_builder.cc")

  5.     list(REMOVE_ITEM _PS_SRC_FILES "optimizer_info.cc")

  6.     list(REMOVE_ITEM _PS_SRC_FILES "scheduler.cc")

  7.     list(REMOVE_ITEM _PS_SRC_FILES "util.cc")

  8.     list(REMOVE_ITEM _PS_SRC_FILES "embedding_table_shard_metadata.cc")

  9.     list(REMOVE_ITEM _PS_SRC_FILES "core/communicator/http_message_handler.cc")

  10.     list(REMOVE_ITEM _PS_SRC_FILES "core/communicator/http_server.cc")

  11.     list(REMOVE_ITEM _PS_SRC_FILES "core/comm_util.cc")

  12.     list(REMOVE_ITEM _PS_SRC_FILES "core/communicator/tcp_client.cc")

  13.     list(REMOVE_ITEM _PS_SRC_FILES "core/communicator/tcp_message_handler.cc")

  14.     list(REMOVE_ITEM _PS_SRC_FILES "core/communicator/tcp_server.cc")

  15.     list(REMOVE_ITEM _PS_SRC_FILES "core/node.cc")

  16.     list(REMOVE_ITEM _PS_SRC_FILES "core/node_manager.cc")

  17.     list(REMOVE_ITEM _PS_SRC_FILES "ps_cache/ps_cache_manager.cc")

  18.     list(REMOVE_ITEM _PS_SRC_FILES "core/worker_node.cc")

  19.     list(REMOVE_ITEM _PS_SRC_FILES "core/server_node.cc")

  20.     list(REMOVE_ITEM _PS_SRC_FILES "core/abstract_node.cc")

  21.     list(REMOVE_ITEM _PS_SRC_FILES "core/scheduler_node.cc")

  22.     list(REMOVE_ITEM _PS_SRC_FILES "core/communicator/http_client.cc")

  23.     list(REMOVE_ITEM _PS_SRC_FILES "worker.cc")

  24.     list(REMOVE_ITEM _PS_SRC_FILES "parameter_server.cc")

  25.     list(REMOVE_ITEM _PS_SRC_FILES "core/communicator/http_request_handler.cc")

  26.     list(REMOVE_ITEM _PS_SRC_FILES "core/communicator/ssl_wrapper.cc")

  27.     list(REMOVE_ITEM _PS_SRC_FILES "core/communicator/ssl_http.cc")

  28.     list(REMOVE_ITEM _PS_SRC_FILES "core/communicator/ssl_client.cc")

  29.     list(REMOVE_ITEM _PS_SRC_FILES "core/leader_scaler.cc")

  30.     list(REMOVE_ITEM _PS_SRC_FILES "core/follower_scaler.cc")

  31.     list(REMOVE_ITEM _PS_SRC_FILES "core/file_configuration.cc")

  32.     list(REMOVE_ITEM _PS_SRC_FILES "core/recovery_base.cc")

  33.     list(REMOVE_ITEM _PS_SRC_FILES "core/node_recovery.cc")

  34.     list(REMOVE_ITEM _PS_SRC_FILES "core/scheduler_recovery.cc")

  35.     list(REMOVE_ITEM _PS_SRC_FILES "core/communicator/communicator_base.cc")

  36.     list(REMOVE_ITEM _PS_SRC_FILES "core/communicator/http_communicator.cc")

  37.     list(REMOVE_ITEM _PS_SRC_FILES "core/communicator/tcp_communicator.cc")

  38.     list(REMOVE_ITEM _PS_SRC_FILES "core/communicator/http_msg_handler.cc")

  39.     list(REMOVE_ITEM _PS_SRC_FILES "core/communicator/tcp_msg_handler.cc")

  40.     list(REMOVE_ITEM _PS_SRC_FILES "core/instance_manager.cc")

  41. endif()

  42. 

  43. if(NOT ENABLE_D)

  44.     list(REMOVE_ITEM _PS_SRC_FILES "ps_cache/ascend/ascend_ps_cache.cc")

  45. endif()

  46. 

  47. if(NOT ENABLE_GPU)

  48.     list(REMOVE_ITEM _PS_SRC_FILES "ps_cache/gpu/gpu_ps_cache.cc")

  49. endif()

  50. 

  51. list(REMOVE_ITEM _PS_SRC_FILES "ps_cache/ps_data/ps_data_prefetch.cc")

  52. list(REMOVE_ITEM _PS_SRC_FILES "ps_cache/ps_data/ps_data_channel.cc")

  53. add_subdirectory(ps_cache)

  54. 

  55. set_property(SOURCE ${_PS_SRC_FILES} PROPERTY COMPILE_DEFINITIONS SUBMODULE_ID=mindspore::SubModuleId::SM_PS)

  56. add_library(_mindspore_ps_obj OBJECT ${_PS_SRC_FILES})

  57. target_link_libraries(_mindspore_ps_obj mindspore::flatbuffers)