MCPcopy Create free account
hub / github.com/catboost/catboost / TMasterContext

Method TMasterContext

catboost/private/libs/distributed/master.cpp:41–56  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

39};
40
41TMasterContext::TMasterContext(const NCatboostOptions::TSystemOptions& systemOptions) {
42 Y_ASSERT(systemOptions.IsMaster());
43 const ui32 unusedNodePort = NCatboostOptions::TSystemOptions::GetUnusedNodePort();
44
45 // avoid Netliba
46 NPar::TParNetworkSettings::GetRef().RequesterType = NPar::TParNetworkSettings::ERequesterType::NEH;
47 TMasterEnvironment::GetRef().RootEnvironment = NPar::RunMaster(
48 systemOptions.NodePort,
49 systemOptions.NumThreads,
50 systemOptions.FileWithHosts->c_str(),
51 unusedNodePort,
52 unusedNodePort);
53 const int workerCount = TMasterEnvironment::GetRef().RootEnvironment->GetSlaveCount();
54 const auto& workerMapping = TMasterEnvironment::GetRef().RootEnvironment->MakeHostIdMapping(workerCount);
55 TMasterEnvironment::GetRef().SharedTrainData = TMasterEnvironment::GetRef().RootEnvironment->CreateEnvironment(SHARED_ID_TRAIN_DATA, workerMapping);
56}
57
58TMasterContext::~TMasterContext() {
59 if (TMasterEnvironment::GetRef().RootEnvironment != nullptr) {

Callers

nothing calls this directly

Calls 6

RunMasterFunction · 0.50
IsMasterMethod · 0.45
c_strMethod · 0.45
GetSlaveCountMethod · 0.45
MakeHostIdMappingMethod · 0.45
CreateEnvironmentMethod · 0.45

Tested by

no test coverage detected