Method FillOrder

catboost/cuda/data/permutation.cpp:7–18

#include <numeric>

void NCatboostCuda::TDataPermutation::FillOrder(TVector<ui32>& order) const {
    if (Index != IdentityPermutationId()) {
        if (DataProvider->MetaInfo.HasGroupId && !DataProvider->ObjectsGrouping->IsTrivial()) {
            GenerateQueryDocsOrder(GetSeed(), BlockSize, DataProvider->ObjectsGrouping->GetNonTrivialGroups(), &order);
        } else {
            Shuffle(GetSeed(), BlockSize, DataProvider->GetObjectCount(), &order);
        }
    } else {
        order.resize(DataProvider->GetObjectCount());
        std::iota(order.begin(), order.end(), 0);
    }
}

void NCatboostCuda::TDataPermutation::FillGroupOrder(TVector<ui32>& groupOrder) const {
    CB_ENSURE_INTERNAL(DataProvider->MetaInfo.HasGroupId, "FillGroupOrder supports only datasets with group ids");