Function StartCudaManager

catboost/cuda/cuda_lib/cuda_manager.cpp:283–302

}

THolder<TStopCudaManagerCallback> StartCudaManager(const NCudaLib::TDeviceRequestConfig& requestConfig,
                                                   const ELoggingLevel loggingLevel) {
    TSetLogging inThisScope(loggingLevel);

#if defined(USE_MPI)
    CB_ENSURE(GetMpiManager().IsMaster(), "Error: can't run cudaManager on slave");
#endif

    auto& manager = NCudaLib::GetCudaManager();
    manager.Start(requestConfig);
    manager.WaitComplete();

    ui32 devCount = manager.GetDeviceCount();
    for (ui32 dev = 0; dev < devCount; ++dev) {
        CATBOOST_INFO_LOG << "Free memory on device #" << dev << " " << manager.FreeMemoryMb(dev) << "MB" << Endl;
    }
    InitMemPerformanceTables(manager);

    return MakeHolder<TStopCudaManagerCallback>();
}

THolder<TStopCudaManagerCallback> StartCudaManager(const ELoggingLevel loggingLevel) {
    return StartCudaManager(NCudaLib::GetDefaultDeviceRequestConfig(), loggingLevel);