| 139 | template <class TDataSet, |
| 140 | class TMapping> |
| 141 | inline void CreateCompressedSplit(const TDataSet& dataSet, |
| 142 | const NCudaLib::TDistributedObject<TCFeature>& feature, |
| 143 | ui32 bin, |
| 144 | TCudaBuffer<ui64, TMapping>& bits, |
| 145 | const TCudaBuffer<const ui32, TMapping>* readIndices = nullptr, |
| 146 | ui32 stream = 0) { |
| 147 | using TKernel = NKernelHost::TWriteCompressedSplitKernel; |
| 148 | |
| 149 | LaunchKernels<TKernel>(bits.NonEmptyDevices(), stream, feature, bin, |
| 150 | dataSet.GetCompressedIndex(), |
| 151 | dataSet.GetSampleCount(), |
| 152 | bits, |
| 153 | readIndices); |
| 154 | } |
| 155 | |
| 156 | template <class TMapping, class TFloat> |
| 157 | inline void CreateCompressedSplitFloat(const TCudaBuffer<TFloat, TMapping>& feature, float border, |
no test coverage detected