@@ -60,54 +60,6 @@ static void k_get_rows(
6060 dst_row[iybs + iqs + y_offset] = v.y ();
6161}
6262
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-
11163template <typename src0_t , typename dst_t >
11264static void k_get_rows_float (
11365 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
177129 GGML_UNUSED (ctx);
178130}
179131
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-
221132template <typename src0_t >
222133static void get_rows_sycl_float (ggml_backend_sycl_context & ctx, const ggml_tensor *src0,
223134 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) {
277188 src1_i32, (float *)dst->data , ctx.stream ());
278189 break ;
279190 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- }
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 ());
287193 break ;
288194 case GGML_TYPE_Q4_1:
289195 get_rows_sycl<QK4_1, QR4_1, dequantize_q4_1>(ctx, dst->src [0 ], dst->src [1 ], dst, (const float *)dst->src [0 ]->data ,
0 commit comments