@@ -23,35 +23,36 @@ PServerUtil::PServerUtil(const ParameterServerConfig& config) {
23
23
int onlineCpus = rdma::numCpus ();
24
24
int numPorts = config.ports_num () + config.ports_num_for_sparse ();
25
25
26
- if (FLAGS_nics .empty ()) {
26
+ if (config. nics () .empty ()) {
27
27
pservers_.resize (numPorts);
28
28
for (int i = 0 ; i < numPorts; ++i) {
29
- if (FLAGS_rdma_tcp == " rdma" ) {
29
+ if (config. rdma_tcp () == " rdma" ) {
30
30
pservers_[i].reset (
31
- new ParameterServer2 (std::string (), FLAGS_port + i, rdmaCpu++));
31
+ new ParameterServer2 (std::string (), config. port () + i, rdmaCpu++));
32
32
rdmaCpu = rdmaCpu % onlineCpus;
33
33
} else {
34
- pservers_[i].reset (new ParameterServer2 (std::string (), FLAGS_port + i));
34
+ pservers_[i].reset (
35
+ new ParameterServer2 (std::string (), config.port () + i));
35
36
}
36
37
CHECK (pservers_[i]->init ()) << " Fail to initialize parameter server"
37
- << FLAGS_port + i;
38
+ << config. port () + i;
38
39
}
39
40
} else {
40
- str::split (FLAGS_nics , ' ,' , &devices);
41
+ str::split (config. nics () , ' ,' , &devices);
41
42
pservers_.resize (devices.size () * numPorts);
42
43
for (int i = 0 ; i < numPorts; ++i) {
43
44
for (size_t j = 0 ; j < devices.size (); ++j) {
44
- if (FLAGS_rdma_tcp == " rdma" ) {
45
+ if (config. rdma_tcp () == " rdma" ) {
45
46
pservers_[i * devices.size () + j].reset (new ParameterServer2 (
46
- getIpAddr (devices[j]), FLAGS_port + i, rdmaCpu++));
47
+ getIpAddr (devices[j]), config. port () + i, rdmaCpu++));
47
48
rdmaCpu = rdmaCpu % onlineCpus;
48
49
} else {
49
50
pservers_[i * devices.size () + j].reset (
50
- new ParameterServer2 (getIpAddr (devices[j]), FLAGS_port + i));
51
+ new ParameterServer2 (getIpAddr (devices[j]), config. port () + i));
51
52
}
52
53
CHECK (pservers_[i * devices.size () + j]->init ())
53
54
<< " Fail to initialize parameter server" << devices[j]
54
- << FLAGS_port + i;
55
+ << config. port () + i;
55
56
}
56
57
}
57
58
}
0 commit comments