fail with exception if user set doc-parallel and has non-power of 2 devices
@@ -82,15 +82,15 @@ namespace NCatboostCuda {
return;
} else {
if (catBoostOptions.BoostingOptions->BoostingType == EBoostingType::Plain) {
+ if (catBoostOptions.BoostingOptions->DataPartitionType.NotSet()) {
if (!couldUsedDocParallelBoosting) {
MATRIXNET_WARNING_LOG << "Can't used doc-parallel data-partition for non-power of 2 device count. It could be more efficient to use less devices, but with doc-parallel mode."<<Endl;
- return;
- }
- if (catBoostOptions.BoostingOptions->DataPartitionType.NotSet()) {
+ } else {
catBoostOptions.BoostingOptions->DataPartitionType = EDataPartitionType::DocParallel;
}
}
}
+ }
if (catBoostOptions.BoostingOptions->DataPartitionType == EDataPartitionType::DocParallel) {
CB_ENSURE(couldUsedDocParallelBoosting, "Can't use doc-parallel data partition for non-power of two device currently. We'll fix it in upcoming release. Please use FeatureParallel mode instead");
