| 317 | } |
| 318 | |
| 319 | CATBOOST_API bool EnableGPUEvaluation(ModelCalcerHandle* modelHandle, int deviceId) { |
| 320 | try { |
| 321 | //TODO(kirillovs): fix this after adding set evaluator props interface |
| 322 | CB_ENSURE(deviceId == 0, "FIXME: Only device 0 is supported for now"); |
| 323 | FULL_MODEL_PTR(modelHandle)->SetEvaluatorType(EFormulaEvaluatorType::GPU); |
| 324 | } catch (...) { |
| 325 | ErrorMessageHolder.Get().Message = CurrentExceptionMessage(); |
| 326 | return false; |
| 327 | } |
| 328 | return true; |
| 329 | } |
| 330 | |
| 331 | CATBOOST_API bool GetSupportedEvaluatorTypes( |
| 332 | ModelCalcerHandle* modelHandle, |
no test coverage detected