Method FreeMemoryMb

catboost/cuda/cuda_lib/cuda_manager.cpp:32–38

}

double TCudaManager::FreeMemoryMb(ui32 devId, bool waitComplete) const {
    if (waitComplete) {
        GetCudaManager().WaitComplete();
    }
    auto devPtr = GetState().Devices[devId];
    return devPtr->GetMemoryState().FreeGpuRam * 1.0 / 1024 / 1024;
}

double TCudaManager::TotalMemoryMb(ui32 devId) const {
    auto devPtr = GetState().Devices[devId];