| 114 | } |
| 115 | template <typename T, typename TIndex> |
| 116 | void MoveSparseValues(const GPUDevice& d, const TIndex* keys, const TIndex* idxs, |
| 117 | const T* values, const int64 size, T* out) { |
| 118 | GpuLaunchConfig config = GetGpuLaunchConfig(size, d); |
| 119 | MoveSparseValuesKernel<<<config.block_count, config.thread_per_block, 0, |
| 120 | d.stream()>>>(keys, idxs, values, size, out); |
| 121 | } |
| 122 | template <typename T, typename TIndex> |
| 123 | void CompareAdjacent(const GPUDevice& d, const T* in, const int64 size, |
| 124 | TIndex* out) { |
no test coverage detected