Function ComputeQueryIds

catboost/cuda/gpu_data/kernels.h:495–502

template <class TUi32, class TMapping, class TQueryOffsetsBias>
inline void ComputeQueryIds(const TCudaBuffer<TUi32, TMapping>& querySizes,
                            const TCudaBuffer<TUi32, TMapping>& queryOffsets,
                            TQueryOffsetsBias bias,
                            TCudaBuffer<ui32, TMapping>* qids,
                            ui32 stream = 0) {
    using TKernel = NKernelHost::TComputeQueryIdsKernel;
    LaunchKernels<TKernel>(querySizes.NonEmptyDevices(), stream, querySizes, queryOffsets, bias, *qids);
};

template <class TUi32, class TMapping>
inline void FillQueryEndMasks(const TCudaBuffer<TUi32, TMapping>& qids,