| 39 | }; |
| 40 | |
| 41 | TMasterContext::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 | |
| 58 | TMasterContext::~TMasterContext() { |
| 59 | if (TMasterEnvironment::GetRef().RootEnvironment != nullptr) { |
nothing calls this directly
no test coverage detected