Method AddDataSetPacks

catboost/cuda/methods/tree_ctrs.cpp:306–348

}

void NCatboostCuda::TTreeCtrDataSetsHelper::AddDataSetPacks(const NCatboostCuda::TFeatureTensor& baseTensor,
                                                            const TSingleBuffer<const ui32>& baseTensorIndices,
                                                            ui32 deviceId,
                                                            TVector<NCatboostCuda::TTreeCtrDataSetsHelper::TTreeCtrDataSetPtr>& dst) {
    const auto& catFeatures = DataSet.GetCatFeatures();
    auto& devFeatures = catFeatures.GetDeviceFeatures(deviceId);
    if (devFeatures.size() == 0) {
        return;
    }
    const ui32 maxPackSize = PackSizeEstimators[deviceId]->GetMaxPackSize();
    CB_ENSURE(maxPackSize, "Error: not enough memory for building ctrs");

    const ui32 currentDstSize = static_cast<const ui32>(dst.size());
    dst.push_back(MakeHolder<TTreeCtrDataSet>(FeaturesManager,
                                              baseTensor,
                                              baseTensorIndices));

    ui32 packSize = 0;
    for (auto feature : devFeatures) {
        auto& nextDataSet = dst.back();
        auto tensor = baseTensor;
        tensor.AddCatFeature(feature);
        if (tensor == baseTensor || !FeaturesManager.UseForTreeCtr(tensor)) {
            continue;
        }
        nextDataSet->AddCatFeature(feature);
        ++packSize;

        if (packSize >= maxPackSize) {
            dst.push_back(MakeHolder<TTreeCtrDataSet>(FeaturesManager,
                                                      baseTensor,
                                                      baseTensorIndices));
            packSize = 0;
        }
    }

    if (dst.back()->CatFeatures.size() == 0) {
        dst.pop_back();
    }
    for (ui32 i = currentDstSize; i < dst.size(); ++i) {
        dst[i]->BuildFeatureIndex();
    }
}

NCatboostCuda::TFeatureTensorTracker NCatboostCuda::TTreeCtrDataSetsHelper::CreateEmptyTrackerForTensor(const NCatboostCuda::TFeatureTensor& tensor) {
    ui64 maxSize = 0;