Method Run

catboost/cuda/methods/pairwise_kernels.cpp:24–48

}

void TMakeLinearSystemKernel::Run(const TCudaStream& stream) const {
    const ui32 systemSize = LinearSystem.ObjectSize();
    const ui32 rowSize = GetRowSizeFromLinearSystemSize(systemSize);
    CB_ENSURE(rowSize > 1, systemSize);
    const ui32 leavesCount = rowSize / 2;

    CB_ENSURE(systemSize == (rowSize + rowSize * (rowSize + 1) / 2));
    CB_ENSURE(BlockFeaturesSlice.Size() <= HistogramLineSize);
    CB_ENSURE(BlockFeaturesSlice.Size() == LinearSystem.ObjectCount());
    const bool useWeights = PointwiseHistogram.ObjectSize() == 2;

    NKernel::MakePointwiseDerivatives(PointwiseHistogram.Get(), HistogramLineSize,
                                      PartStats.Get(),
                                      useWeights,
                                      rowSize,
                                      BlockFeaturesSlice.Left, BlockFeaturesSlice.Size(),
                                      LinearSystem.Get(), stream.GetStream());

    NKernel::MakePairwiseDerivatives(PairwiseHistogram.Get(),
                                     leavesCount,
                                     BlockFeaturesSlice.Left, BlockFeaturesSlice.Size(),
                                     HistogramLineSize,
                                     LinearSystem.Get(),
                                     stream.GetStream());
}

void TExtractMatricesAndTargetsKernel::Run(const TCudaStream& stream) const {
    const ui32 rowSize = GetRowSizeFromLinearSystemSize(LinearSystem.ObjectSize());