Skip to content

Commit 5aaaef4

Browse files
committed
rename PServerController to ParameterServerController
1 parent 3f6c2b3 commit 5aaaef4

File tree

5 files changed

+31
-28
lines changed

5 files changed

+31
-28
lines changed

paddle/pserver/CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,14 +25,14 @@ set(PSERVER_SOURCES
2525
ParameterClient2.cpp
2626
ParameterServer2.cpp
2727
SparseParameterDistribution.cpp
28-
PServerController.cpp)
28+
ParameterServerController.cpp)
2929

3030
set(PSERVER_HEADERS
3131
BaseClient.h
3232
ParameterClient2.h
3333
ParameterServer2.h
3434
SparseParameterDistribution.h
35-
PServerController.h)
35+
ParameterServerController.h)
3636

3737
add_library(paddle_pserver STATIC
3838
${PSERVER_SOURCES})

paddle/pserver/ParameterServer2Main.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,15 +13,15 @@ See the License for the specific language governing permissions and
1313
limitations under the License. */
1414

1515
#include <fstream>
16-
#include "PServerController.h"
16+
#include "ParameterServerController.h"
1717

1818
using namespace paddle; // NOLINT
1919

2020
int main(int argc, char** argv) {
2121
initMain(argc, argv);
2222

23-
std::unique_ptr<PServerController> pServerPtr(
24-
paddle::PServerController::createByGflags());
23+
std::unique_ptr<ParameterServerController> pServerPtr(
24+
paddle::ParameterServerController::createByGflags());
2525
pServerPtr->start();
2626
pServerPtr->join();
2727

paddle/pserver/PServerController.cpp renamed to paddle/pserver/ParameterServerController.cpp

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,12 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1212
See the License for the specific language governing permissions and
1313
limitations under the License. */
1414

15-
#include "PServerController.h"
15+
#include "ParameterServerController.h"
1616

1717
namespace paddle {
1818

19-
PServerController::PServerController(const ParameterServerConfig& config) {
19+
ParameterServerController::ParameterServerController(
20+
const ParameterServerConfig& config) {
2021
// round robin to load balance RDMA server ENGINE
2122
std::vector<std::string> devices;
2223
int rdmaCpu = 0;
@@ -58,9 +59,9 @@ PServerController::PServerController(const ParameterServerConfig& config) {
5859
}
5960
}
6061

61-
PServerController::~PServerController() { this->join(); }
62+
ParameterServerController::~ParameterServerController() { this->join(); }
6263

63-
PServerController* PServerController::createByGflags() {
64+
ParameterServerController* ParameterServerController::createByGflags() {
6465
ParameterServerConfig config;
6566

6667
config.set_nics(FLAGS_nics);
@@ -72,12 +73,12 @@ PServerController* PServerController::createByGflags() {
7273
return create(config);
7374
}
7475

75-
PServerController* PServerController::create(
76+
ParameterServerController* ParameterServerController::create(
7677
const ParameterServerConfig& config) {
77-
return new PServerController(config);
78+
return new ParameterServerController(config);
7879
}
7980

80-
void PServerController::start() {
81+
void ParameterServerController::start() {
8182
LOG(INFO) << "pserver sizes : " << pservers_.size();
8283
int i = 0;
8384
for (const auto& pserver : pservers_) {
@@ -87,7 +88,7 @@ void PServerController::start() {
8788
}
8889
}
8990

90-
void PServerController::join() {
91+
void ParameterServerController::join() {
9192
LOG(INFO) << "pserver sizes : " << pservers_.size();
9293
int i = 0;
9394
for (const auto& pserver : pservers_) {

paddle/pserver/PServerController.h renamed to paddle/pserver/ParameterServerController.h

Lines changed: 14 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -21,39 +21,41 @@ limitations under the License. */
2121

2222
namespace paddle {
2323

24-
class PServerController final {
24+
class ParameterServerController final {
2525
public:
26-
DISABLE_COPY(PServerController);
26+
DISABLE_COPY(ParameterServerController);
2727

2828
/**
29-
* @brief Ctor, Create a PServerUtil from ParameterServerConfig.
29+
* @brief Ctor, Create a ParameterServerController from ParameterServerConfig.
3030
*/
31-
explicit PServerController(const ParameterServerConfig& config);
31+
explicit ParameterServerController(const ParameterServerConfig& config);
3232

3333
/**
3434
* @brief Dtor.
3535
*/
36-
~PServerController();
36+
~ParameterServerController();
3737

3838
/**
39-
* @brief create PServerUtil from gflags, this is used for
39+
* @brief create ParameterServerController from gflags, this is used for
4040
* compatibility with the old usage of configuration by gflags.
4141
*/
42-
static PServerController* createByGflags();
42+
static ParameterServerController* createByGflags();
4343

4444
/**
45-
* @brief create PServerUtil with ParameterServerConfig, remove gflags
46-
* from ParameterServer. Init all pservers thread according to the config.
45+
* @brief create ParameterServerController with ParameterServerConfig, remove
46+
* gflags from ParameterServer. Init all pservers thread according to the
47+
* config.
4748
*/
48-
static PServerController* create(const ParameterServerConfig& config);
49+
static ParameterServerController* create(const ParameterServerConfig& config);
4950

5051
/**
51-
* @brief start all pserver thread in this PServerUtil.
52+
* @brief start all pserver thread in this ParameterServerController.
5253
*/
5354
void start();
5455

5556
/**
56-
* @brief join and wait for all pserver thread in this PServerUtil.
57+
* @brief join and wait for all pserver thread in this
58+
* ParameterServerController.
5759
*/
5860
void join();
5961

paddle/trainer/TrainerMain.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
1313
limitations under the License. */
1414

1515
#include <fenv.h>
16-
#include "paddle/pserver/PServerController.h"
16+
#include "paddle/pserver/ParameterServerController.h"
1717
#include "paddle/utils/Excepts.h"
1818
#include "paddle/utils/PythonUtil.h"
1919

@@ -37,9 +37,9 @@ int main(int argc, char** argv) {
3737
initMain(argc, argv);
3838
initPython(argc, argv);
3939

40-
std::unique_ptr<PServerController> pServerPtr(nullptr);
40+
std::unique_ptr<ParameterServerController> pServerPtr(nullptr);
4141
if (FLAGS_start_pserver) {
42-
pServerPtr.reset(paddle::PServerController::createByGflags());
42+
pServerPtr.reset(paddle::ParameterServerController::createByGflags());
4343
pServerPtr->start();
4444
}
4545
Trainer trainer;

0 commit comments

Comments
 (0)