Function CreateLocalExecutor

catboost/libs/train_lib/train_model.cpp:61–71

using namespace NCB;

static THolder<NPar::ILocalExecutor> CreateLocalExecutor(const NCatboostOptions::TCatBoostOptions& catBoostOptions) {
    const bool isGpuDeviceType = catBoostOptions.GetTaskType() == ETaskType::GPU;
    const int threadCount = catBoostOptions.SystemOptions.Get().NumThreads.Get();
    if (isGpuDeviceType && TTrainerFactory::Has(ETaskType::GPU)) {
        auto localExecutorHolder = MakeHolder<NPar::TLocalExecutor>();
        localExecutorHolder->RunAdditionalThreads(threadCount - 1);
        return localExecutorHolder;
    } else {
        return MakeHolder<NPar::TTbbLocalExecutor<>>(threadCount);
    }
}

static void ShrinkModel(int itCount, const TCtrHelper& ctrsHelper, TLearnProgress* progress) {
    itCount += SafeIntegerCast<int>(progress->InitTreesSize);