Anton Mitkov commited on
Commit
feee739
·
1 Parent(s): 26bafb6

sycl: Cleanup codepaths in Get Rows in sycl backend (llama/14215)

Browse files
Files changed (1) hide show
  1. ggml/src/ggml-sycl/getrows.cpp +2 -96
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
- if (ctx.opt_feature.reorder && dst->op == GGML_OP_MUL_MAT) {
281
- get_rows_sycl_reorder<QK4_0, QR4_0, dequantize_q4_0_reorder>(ctx, dst->src[0], dst->src[1], dst, (const float *)dst->src[0]->data,
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,