| 57 | } |
| 58 | template <typename TIndex> |
| 59 | __global__ void MoveValuesKernel(const TIndex* keys, const TIndex* values, |
| 60 | const int64 size, TIndex* out) { |
| 61 | GPU_1D_KERNEL_LOOP(i, size) { |
| 62 | TIndex key = ldg(keys + i); |
| 63 | out[key] = ldg(values + i); |
| 64 | } |
| 65 | } |
| 66 | template <typename TIndex> |
| 67 | __global__ void MoveValuesKernel(const TIndex* keys, const TIndex* values, |
| 68 | const int64* size_ptr, TIndex* out) { |