Spaces:
Running
Running
Anton Mitkov
commited on
Commit
·
feee739
1
Parent(s):
26bafb6
sycl: Cleanup codepaths in Get Rows in sycl backend (llama/14215)
Browse files
ggml/src/ggml-sycl/getrows.cpp
CHANGED
|
@@ -60,54 +60,6 @@ static void k_get_rows(
|
|
| 60 |
dst_row[iybs + iqs + y_offset] = v.y();
|
| 61 |
}
|
| 62 |
|
| 63 |
-
template<int qk, int qr, dequantize_kernel_t_reorder dequantize_kernel_recorder, typename dst_t>
|
| 64 |
-
static void k_get_rows_reorder(
|
| 65 |
-
const void * src0, const void *src0_dq, const int32_t * src1, dst_t * dst,
|
| 66 |
-
int64_t ne00, /*int64_t ne01, int64_t ne02, int64_t ne03,*/
|
| 67 |
-
/*int64_t ne10, int64_t ne11,*/ int64_t ne12, /*int64_t ne13,*/
|
| 68 |
-
/*size_t s0,*/ size_t s1, size_t s2, size_t s3,
|
| 69 |
-
/*size_t nb00,*/ size_t nb01, size_t nb02, size_t nb03,
|
| 70 |
-
size_t s10, size_t s11, size_t s12,
|
| 71 |
-
const sycl::nd_item<3> &item_ct1/*, size_t s13*/) {
|
| 72 |
-
|
| 73 |
-
const int i00 = (item_ct1.get_group(2) * item_ct1.get_local_range(2) +
|
| 74 |
-
item_ct1.get_local_id(2)) *
|
| 75 |
-
2;
|
| 76 |
-
const int i10 = item_ct1.get_local_range(1) * item_ct1.get_group(1) +
|
| 77 |
-
item_ct1.get_local_id(1);
|
| 78 |
-
const int i11 = (item_ct1.get_group(0) * item_ct1.get_local_range(0) +
|
| 79 |
-
item_ct1.get_local_id(0)) /
|
| 80 |
-
ne12;
|
| 81 |
-
const int i12 = (item_ct1.get_group(0) * item_ct1.get_local_range(0) +
|
| 82 |
-
item_ct1.get_local_id(0)) %
|
| 83 |
-
ne12;
|
| 84 |
-
|
| 85 |
-
if (i00 >= ne00) {
|
| 86 |
-
return;
|
| 87 |
-
}
|
| 88 |
-
auto ncols = ne00;
|
| 89 |
-
const int i01 = src1[i10*s10 + i11*s11 + i12*s12];
|
| 90 |
-
|
| 91 |
-
dst_t * dst_row = dst + i10*s1 + i11*s2 + i12*s3;
|
| 92 |
-
|
| 93 |
-
const int src0_off = i01 * ncols + i00;
|
| 94 |
-
const int ib = src0_off / QK4_0; // block index
|
| 95 |
-
const int iqs = (i00%qk)/qr; // x quant index
|
| 96 |
-
const int iybs = i00 - i00%qk; // dst block start index
|
| 97 |
-
const int y_offset = qr == 1 ? 1 : qk/2;
|
| 98 |
-
|
| 99 |
-
// dequantize
|
| 100 |
-
dfloat2 v;
|
| 101 |
-
dequantize_kernel_recorder((const void *)src0_dq, ib, (const void *)src0, src0_off/2, v);
|
| 102 |
-
|
| 103 |
-
dst_row[iybs + iqs + 0] = v.x();
|
| 104 |
-
dst_row[iybs + iqs + y_offset] = v.y();
|
| 105 |
-
|
| 106 |
-
GGML_UNUSED(nb01);
|
| 107 |
-
GGML_UNUSED(nb02);
|
| 108 |
-
GGML_UNUSED(nb03);
|
| 109 |
-
}
|
| 110 |
-
|
| 111 |
template<typename src0_t, typename dst_t>
|
| 112 |
static void k_get_rows_float(
|
| 113 |
const src0_t * src0, const int32_t * src1, dst_t * dst,
|
|
@@ -177,47 +129,6 @@ static void get_rows_sycl(ggml_backend_sycl_context & ctx, const ggml_tensor *sr
|
|
| 177 |
GGML_UNUSED(ctx);
|
| 178 |
}
|
| 179 |
|
| 180 |
-
template <int qk, int qr, dequantize_kernel_t_reorder dq_reorder>
|
| 181 |
-
static void get_rows_sycl_reorder(ggml_backend_sycl_context & ctx, const ggml_tensor *src0, const ggml_tensor *src1,
|
| 182 |
-
ggml_tensor *dst, const void *src0_dd,
|
| 183 |
-
const int32_t *src1_dd, float *dst_dd,
|
| 184 |
-
queue_ptr stream) {
|
| 185 |
-
|
| 186 |
-
GGML_TENSOR_BINARY_OP_LOCALS
|
| 187 |
-
|
| 188 |
-
const sycl::range<3> block_dims(1, 1, SYCL_GET_ROWS_BLOCK_SIZE);
|
| 189 |
-
const int block_num_x = (ne00 + 2*SYCL_GET_ROWS_BLOCK_SIZE - 1) / (2*SYCL_GET_ROWS_BLOCK_SIZE);
|
| 190 |
-
const sycl::range<3> block_nums(ne11 * ne12, ne10, block_num_x);
|
| 191 |
-
|
| 192 |
-
// strides in elements
|
| 193 |
-
//const size_t s0 = nb0 / ggml_element_size(dst);
|
| 194 |
-
const size_t s1 = nb1 / ggml_element_size(dst);
|
| 195 |
-
const size_t s2 = nb2 / ggml_element_size(dst);
|
| 196 |
-
const size_t s3 = nb3 / ggml_element_size(dst);
|
| 197 |
-
|
| 198 |
-
const size_t s10 = nb10 / ggml_element_size(src1);
|
| 199 |
-
const size_t s11 = nb11 / ggml_element_size(src1);
|
| 200 |
-
const size_t s12 = nb12 / ggml_element_size(src1);
|
| 201 |
-
//const size_t s13 = nb13 / ggml_element_size(src1);
|
| 202 |
-
|
| 203 |
-
GGML_ASSERT(ne00 % 2 == 0);
|
| 204 |
-
|
| 205 |
-
const uint8_t* src0_q = (const uint8_t*)src0_dd;
|
| 206 |
-
const size_t ncols = ne00;
|
| 207 |
-
const size_t nrows = ne01;
|
| 208 |
-
const sycl::half* src0_dq = (const sycl::half*)(src0_q + nrows * ncols / 2);
|
| 209 |
-
stream->parallel_for(sycl::nd_range<3>(block_nums * block_dims, block_dims),
|
| 210 |
-
[=](sycl::nd_item<3> item_ct1) [[sycl::reqd_sub_group_size(WARP_SIZE)]]{
|
| 211 |
-
k_get_rows_reorder<qk, qr, dq_reorder>(
|
| 212 |
-
src0_dd, src0_dq, src1_dd, dst_dd, ne00, ne12, s1, s2,
|
| 213 |
-
s3, nb01, nb02, nb03, s10, s11, s12, item_ct1);
|
| 214 |
-
});
|
| 215 |
-
|
| 216 |
-
GGML_UNUSED(dst);
|
| 217 |
-
GGML_UNUSED(ctx);
|
| 218 |
-
}
|
| 219 |
-
|
| 220 |
-
|
| 221 |
template <typename src0_t>
|
| 222 |
static void get_rows_sycl_float(ggml_backend_sycl_context & ctx, const ggml_tensor *src0,
|
| 223 |
const ggml_tensor *src1, ggml_tensor *dst,
|
|
@@ -277,13 +188,8 @@ void ggml_sycl_op_get_rows(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
|
| 277 |
src1_i32, (float *)dst->data, ctx.stream());
|
| 278 |
break;
|
| 279 |
case GGML_TYPE_Q4_0:
|
| 280 |
-
|
| 281 |
-
|
| 282 |
-
src1_i32, (float *)dst->data, ctx.stream());
|
| 283 |
-
} else {
|
| 284 |
-
get_rows_sycl<QK4_0, QR4_0, dequantize_q4_0>(ctx, dst->src[0], dst->src[1], dst, (const float *)dst->src[0]->data,
|
| 285 |
-
src1_i32, (float *)dst->data, ctx.stream());
|
| 286 |
-
}
|
| 287 |
break;
|
| 288 |
case GGML_TYPE_Q4_1:
|
| 289 |
get_rows_sycl<QK4_1, QR4_1, dequantize_q4_1>(ctx, dst->src[0], dst->src[1], dst, (const float *)dst->src[0]->data,
|
|
|
|
| 60 |
dst_row[iybs + iqs + y_offset] = v.y();
|
| 61 |
}
|
| 62 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 63 |
template<typename src0_t, typename dst_t>
|
| 64 |
static void k_get_rows_float(
|
| 65 |
const src0_t * src0, const int32_t * src1, dst_t * dst,
|
|
|
|
| 129 |
GGML_UNUSED(ctx);
|
| 130 |
}
|
| 131 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 132 |
template <typename src0_t>
|
| 133 |
static void get_rows_sycl_float(ggml_backend_sycl_context & ctx, const ggml_tensor *src0,
|
| 134 |
const ggml_tensor *src1, ggml_tensor *dst,
|
|
|
|
| 188 |
src1_i32, (float *)dst->data, ctx.stream());
|
| 189 |
break;
|
| 190 |
case GGML_TYPE_Q4_0:
|
| 191 |
+
get_rows_sycl<QK4_0, QR4_0, dequantize_q4_0>(ctx, dst->src[0], dst->src[1], dst, (const float *)dst->src[0]->data,
|
| 192 |
+
src1_i32, (float *)dst->data, ctx.stream());
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 193 |
break;
|
| 194 |
case GGML_TYPE_Q4_1:
|
| 195 |
get_rows_sycl<QK4_1, QR4_1, dequantize_q4_1>(ctx, dst->src[0], dst->src[1], dst, (const float *)dst->src[0]->data,
|