Home
last modified time | relevance | path

Searched defs:server_num (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/ps/core/
Dcluster_metadata.h36 uint32_t server_num; member
Dnode_recovery.cc39 …int32_t server_num = std::strtol(recovery_storage_->Get(kRecoveryServerNum, "").c_str(), nullptr, … in Recover() local
Dnode_manager.cc377 void NodeManager::set_server_num(const uint32_t &server_num) { meta_data_->server_num = server_num;… in set_server_num()
381 uint32_t NodeManager::server_num() const { return meta_data_->server_num; } in server_num() function in mindspore::ps::core::NodeManager
Dabstract_node.cc576 uint32_t AbstractNode::server_num() const { return server_num_; } in server_num() function in mindspore::ps::core::AbstractNode
580 void AbstractNode::set_server_num(const uint32_t &server_num) { server_num_ = server_num; } in set_server_num()
952 int32_t server_num = scale_out_message.server_num(); in ProcessScaleOut() local
972 int32_t server_num = scale_in_message.server_num(); in ProcessScaleIn() local
Dps_scheduler_node.cc48 uint32_t server_num = clusterConfig.initial_server_num; in RunRecovery() local
Dscheduler_node.cc114 uint32_t server_num = clusterConfig.initial_server_num; in RunRecovery() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ps/core/
Dcluster_metadata_test.cc38 std::string server_num = "1"; in TEST_F() local
Dcluster_available_timeout_test.cc36 std::string server_num = "1"; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/ps/
Dutil.cc79 int64_t Util::LocalShard(int64_t first_dim, int64_t rank_id, int64_t server_num) { in LocalShard()
87 …int64_t, int64_t> Util::AllRankLocalShard(int64_t first_dim, int64_t rank_id, int64_t server_num) { in AllRankLocalShard()
Dps_context.cc263 void PSContext::set_server_num(uint32_t server_num) { server_num_ = server_num; } in set_server_num()
264 uint32_t PSContext::server_num() const { return server_num_; } in server_num() function in mindspore::ps::PSContext
/third_party/mindspore/mindspore-src/source/config/
Dhccl_evb_multi_rank.json9 "server_num": "1", string
Dhccl_single_machine_multi_rank.json8 "server_num": "1", string
98 "server_num": "single machine is 1", string
Dhccl_multi_machine_multi_rank.json8 "server_num": "2", string
162 "server_num": "number of multiple machines, single machine is 1", string
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/ps/core/protos/
Dcomm.proto205 uint32 server_num = 3; field
227 uint32 server_num = 2; field
235 uint32 server_num = 2; field
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/
Ddevice_sparse_embedding_operation.cc136 …size_t vocab_size, size_t server_num, std::vector<std::pair<size_t, size_t>> *remote_embedding_sli… in GetRemoteEmbeddingSliceBound() argument
Ddevice_dense_embedding_operation.cc192 …size_t vocab_size, size_t server_num, std::vector<std::pair<size_t, size_t>> *remote_embedding_sli… in GetRemoteEmbeddingSliceBound() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/distributed/embedding_cache/
Dembedding_cache_utils.cc203 auto server_num = ps::PSContext::instance()->server_num(); in GetEmbeddingTableSliceBound() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dgraph_splitter.cc633 uint32_t server_num = ClusterContext::instance()->node_num(distributed::kEnvRoleOfServer); in PostBuildDistributedGraph() local
678 uint32_t server_num = ClusterContext::instance()->node_num(distributed::kEnvRoleOfServer); in PostBuildDistributedGraph() local
742 uint32_t server_num = ClusterContext::instance()->node_num(distributed::kEnvRoleOfServer); in ProcessForSplitOptimizer() local