Home
last modified time | relevance | path

Searched refs:cluster_config (Results 1 – 15 of 15) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/ps/
Dscheduler.cc23 PSContext::instance()->cluster_config().scheduler_host = PSContext::instance()->scheduler_host(); in Run()
24 PSContext::instance()->cluster_config().scheduler_port = PSContext::instance()->scheduler_port(); in Run()
25 …PSContext::instance()->cluster_config().initial_worker_num = PSContext::instance()->initial_worker… in Run()
26 …PSContext::instance()->cluster_config().initial_server_num = PSContext::instance()->initial_server… in Run()
Dps_context.h165 core::ClusterConfig &cluster_config();
Dps_context.cc410 core::ClusterConfig &PSContext::cluster_config() { in cluster_config() function in mindspore::ps::PSContext
/third_party/mindspore/mindspore/ccsrc/ps/core/
Dnode_recovery.cc32 node_->set_worker_num(PSContext::instance()->cluster_config().initial_worker_num); in Recover()
40 node_->set_server_num(PSContext::instance()->cluster_config().initial_server_num); in Recover()
48 node_->set_scheduler_ip(PSContext::instance()->cluster_config().scheduler_host); in Recover()
56 node_->set_scheduler_port(PSContext::instance()->cluster_config().scheduler_port); in Recover()
Dnode_manager.cc23 initial_total_node_num_ = PSContext::instance()->cluster_config().initial_server_num + in InitNode()
24 PSContext::instance()->cluster_config().initial_worker_num; in InitNode()
25 …meta_data_ = std::make_unique<ClusterMetadata>(PSContext::instance()->cluster_config().initial_wor… in InitNode()
26 … PSContext::instance()->cluster_config().initial_server_num); in InitNode()
171 …if (it->second.tv_sec + PSContext::instance()->cluster_config().heartbeat_timeout < current_time.t… in UpdateCluster()
198 << PSContext::instance()->cluster_config().cluster_available_timeout in CheckClusterTimeout()
Dabstract_node.cc487 PSContext::instance()->cluster_config().scheduler_host = scheduler_ip; in GetOrCreateTcpComm()
488 PSContext::instance()->cluster_config().scheduler_port = static_cast<uint16_t>(scheduler_port); in GetOrCreateTcpComm()
489 PSContext::instance()->cluster_config().initial_worker_num = worker_num; in GetOrCreateTcpComm()
490 PSContext::instance()->cluster_config().initial_server_num = server_num; in GetOrCreateTcpComm()
492 PSContext::instance()->cluster_config().scheduler_host = scheduler_ip; in GetOrCreateTcpComm()
493 PSContext::instance()->cluster_config().scheduler_port = static_cast<uint16_t>(scheduler_port); in GetOrCreateTcpComm()
494 PSContext::instance()->cluster_config().initial_worker_num = worker_num; in GetOrCreateTcpComm()
495 PSContext::instance()->cluster_config().initial_server_num = server_num; in GetOrCreateTcpComm()
527 …std::this_thread::sleep_for(std::chrono::seconds(PSContext::instance()->cluster_config().heartbeat… in StartHeartbeatTimer()
561 …int64_t old_time = scheduler_time_.tv_sec + PSContext::instance()->cluster_config().scheduler_time… in CheckSchedulerTimeout()
[all …]
Dworker_node.h40 …bool Start(const uint32_t &timeout = PSContext::instance()->cluster_config().cluster_available_tim…
Dserver_node.h47 …bool Start(const uint32_t &timeout = PSContext::instance()->cluster_config().cluster_available_tim…
Dnode.h61 …virtual bool Start(const uint32_t &timeout = PSContext::instance()->cluster_config().cluster_avail…
Dscheduler_node.cc117 std::string scheduler_host = PSContext::instance()->cluster_config().scheduler_host; in CreateTcpServer()
118 uint32_t scheduler_port = PSContext::instance()->cluster_config().scheduler_port; in CreateTcpServer()
446 … std::chrono::seconds(PSContext::instance()->cluster_config().cluster_available_timeout))) { in StartUpdateClusterStateTimer()
449 …std::this_thread::sleep_for(std::chrono::seconds(PSContext::instance()->cluster_config().heartbeat… in StartUpdateClusterStateTimer()
454 …std::chrono::seconds(PSContext::instance()->cluster_config().heartbeat_interval * kHeartbeatTimes)… in StartUpdateClusterStateTimer()
Dscheduler_node.h67 …bool Start(const uint32_t &timeout = PSContext::instance()->cluster_config().cluster_available_tim…
/third_party/mindspore/tests/ut/cpp/ps/core/
Dcluster_available_timeout_test.cc44 PSContext::instance()->cluster_config().cluster_available_timeout = 3; in TEST_F()
45 …MS_LOG(INFO) << "The timeout is:" << PSContext::instance()->cluster_config().cluster_available_tim… in TEST_F()
Dcluster_metadata_test.cc46 EXPECT_EQ(300, PSContext::instance()->cluster_config().cluster_available_timeout); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/fl/worker/
Dfl_worker.cc37 ps::PSContext::instance()->cluster_config().scheduler_host = scheduler_ip_; in Run()
38 ps::PSContext::instance()->cluster_config().scheduler_port = scheduler_port_; in Run()
39 ps::PSContext::instance()->cluster_config().initial_worker_num = worker_num_; in Run()
40 ps::PSContext::instance()->cluster_config().initial_server_num = server_num_; in Run()
/third_party/mindspore/mindspore/ccsrc/ps/core/communicator/
Dtcp_client.h64 …const uint32_t &connected_timeout = PSContext::instance()->cluster_config().cluster_available_time…