Home
last modified time | relevance | path

Searched refs:server_num (Results 1 – 25 of 64) sorted by relevance

123

/third_party/mindspore/mindspore/ccsrc/ps/
Dutil.cc75 int64_t Util::LocalShard(int64_t first_dim, int64_t rank_id, int64_t server_num) { in LocalShard() argument
76 std::map<int64_t, int64_t> shard_dims = AllRankLocalShard(first_dim, rank_id, server_num); in LocalShard()
83 …int64_t, int64_t> Util::AllRankLocalShard(int64_t first_dim, int64_t rank_id, int64_t server_num) { in AllRankLocalShard() argument
84 if (first_dim <= 0 || server_num <= 0 || rank_id < 0) { in AllRankLocalShard()
87 if (rank_id >= server_num) { in AllRankLocalShard()
88 …PTION) << "The rank ID " << rank_id << " should be less than the number of servers " << server_num; in AllRankLocalShard()
91 for (int64_t i = 0; i < server_num; i++) { in AllRankLocalShard()
94 if (server_num != static_cast<int64_t>(shard_dims.size())) { in AllRankLocalShard()
95 …MS_LOG(EXCEPTION) << "Inconsistent server num " << server_num << " shard dims counter size " << sh… in AllRankLocalShard()
99 server_index = (server_index + 1) % server_num; in AllRankLocalShard()
[all …]
/third_party/mindspore/tests/st/fl/hybrid_lenet/
Drun_smlt.sh18 server_num=$1
25 ofs=`expr $i % $server_num`
28 …python simulator.py --pid=$i --http_ip=$ip --http_port=$port --use_elb=True --server_num=$1 > simu…
Drun_hybrid_train_server.py57 server_num = args.server_num variable
84 local_server_num = server_num
86 assert local_server_num <= server_num, "The local server number should not be bigger than total ser…
99 cmd_server += " --server_num=" + str(server_num)
Drun_hybrid_train_sched.py38 server_num = args.server_num variable
56 cmd_sched += " --server_num=" + str(server_num)
Drun_hybrid_train_worker.py40 server_num = args.server_num variable
67 cmd_worker += " --server_num=" + str(server_num)
/third_party/mindspore/tests/st/fl/mobile/
Drun_smlt.sh18 server_num=$1
25 ofs=`expr $i % $server_num`
28 …python simulator.py --pid=$i --http_ip=$ip --http_port=$port --use_elb=True --server_num=$1 > simu…
Drun_mobile_server.py58 server_num = args.server_num variable
85 local_server_num = server_num
87 …assert local_server_num <= server_num, "The local server number should not be bigger than total se…
100 cmd_server += " --server_num=" + str(server_num)
Drun_mobile_sched.py40 server_num = args.server_num variable
59 cmd_sched += " --server_num=" + str(server_num)
/third_party/mindspore/tests/st/fl/cross_silo_lenet/
Drun_cross_silo_lenet_server.py44 server_num = args.server_num variable
62 local_server_num = server_num
64 assert local_server_num <= server_num, "The local server number should not be bigger than total ser…
76 cmd_server += " --server_num=" + str(server_num)
Drun_cross_silo_lenet_sched.py33 server_num = args.server_num variable
47 cmd_sched += " --server_num=" + str(server_num)
Drun_cross_silo_lenet_worker.py38 server_num = args.server_num variable
65 cmd_worker += " --server_num=" + str(server_num)
/third_party/mindspore/tests/st/fl/cross_silo_femnist/
Drun_cross_silo_femnist_server.py45 server_num = args.server_num variable
64 local_server_num = server_num
66 assert local_server_num <= server_num, "The local server number should not be bigger than total ser…
78 cmd_server += " --server_num=" + str(server_num)
Drun_cross_silo_femnist_sched.py34 server_num = args.server_num variable
49 cmd_sched += " --server_num=" + str(server_num)
Drun_cross_silo_femnist_worker.py39 server_num = args.server_num variable
66 cmd_worker += " --server_num=" + str(server_num)
/third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/
Drun_cross_silo_fasterrcnn_server.py46 server_num = args.server_num variable
66 local_server_num = server_num
68 assert local_server_num <= server_num, "The local server number should not be bigger than total ser…
80 cmd_server += " --server_num=" + str(server_num)
Drun_cross_silo_fasterrcnn_sched.py34 server_num = args.server_num variable
49 cmd_sched += " --server_num=" + str(server_num)
Drun_cross_silo_fasterrcnn_worker.py37 server_num = args.server_num variable
62 cmd_worker += " --server_num=" + str(server_num)
/third_party/mindspore/tests/st/fl/albert/
Drun_hybrid_train_server.py57 server_num = args.server_num variable
83 local_server_num = server_num
85 assert local_server_num <= server_num, "The local server number should not be bigger than total ser…
98 cmd_server += " --server_num=" + str(server_num)
Drun_hybrid_train_sched.py37 server_num = args.server_num variable
54 cmd_sched += " --server_num=" + str(server_num)
/third_party/mindspore/mindspore/ccsrc/ps/core/
Dnode_recovery.cc37 …int32_t server_num = std::strtol(recovery_storage_->Get(kRecoveryServerNum, "").c_str(), nullptr, … in Recover() local
38 node_->set_server_num(server_num); in Recover()
58 …LOG(INFO) << "The worker num:" << node_->worker_num() << ", the server num:" << node_->server_num() in Recover()
Dcluster_metadata.h34 …ClusterMetadata(const uint32_t &worker, const uint32_t &server) : worker_num(worker), server_num(s… in ClusterMetadata()
37 uint32_t server_num; member
Dnode_manager.cc72 if (rank_id >= meta_data_->server_num) { in NextRankId()
73 … MS_LOG(WARNING) << "The rank id is greater than the number of servers:" << meta_data_->server_num; in NextRankId()
306 …oid NodeManager::set_server_num(const int32_t &server_num) { meta_data_->server_num = IntToUint(se… in set_server_num() argument
310 int32_t NodeManager::server_num() const { return UintToInt(meta_data_->server_num); } in server_num() function in mindspore::ps::core::NodeManager
Dcluster_config.h34 …explicit ClusterConfig(const uint32_t &worker_num, const uint32_t &server_num, std::string host, c… in ClusterConfig()
36 initial_server_num(server_num), in ClusterConfig()
/third_party/mindspore/tests/ut/cpp/ps/core/
Dcluster_metadata_test.cc38 std::string server_num = "1"; in TEST_F() local
42 common::SetEnv(kEnvPServerNum, server_num.c_str()); in TEST_F()
Dcluster_available_timeout_test.cc36 std::string server_num = "1"; in TEST_F() local
40 common::SetEnv(kEnvPServerNum, server_num.c_str()); in TEST_F()

123