#pragma clang diagnostic ignored "-Wunused-variable" #pragma clang diagnostic ignored "-Wunused-function" #pragma clang diagnostic ignored "-Wunused-but-set-variable" #include #include #include #include #include #define GGML_COMMON_DECL_C #include "ggml-common.h" #include "hex-cpy-dma.h" #include "htp-ctx.h" #include "htp-fence.h" #include "htp-ops.h" #include "htp-tensor.h" #include "hvx-utils.h" struct htp_copy_context { struct htp_ops_context * octx; uint32_t src0_type_size; uint32_t src0_block_size; uint32_t dst_type_size; uint32_t dst_block_size; uint32_t src0_blocks_per_row; uint32_t dst_blocks_per_row; uint32_t elem_start; uint32_t nelem; uint32_t elem_per_thread; uint32_t src0_nrows_per_thread; uint32_t row_start; uint32_t nrows; struct fastdiv_values div_ne01; struct fastdiv_values div_ne02_ne01; struct fastdiv_values div_ne0; struct fastdiv_values div_ne1_ne0; struct fastdiv_values div_ne2_ne1_ne0; struct fastdiv_values div_ne00; struct fastdiv_values div_ne01_ne00; struct fastdiv_values div_ne02_ne01_ne00; }; #define cpy_preamble \ const struct htp_tensor *src0 = octx->src[0]; \ const struct htp_tensor *dst = octx->dst; \ \ const uint32_t ne00 = src0->ne[0]; \ const uint32_t ne01 = src0->ne[1]; \ const uint32_t ne02 = src0->ne[2]; \ const uint32_t ne03 = src0->ne[3]; \ \ const uint32_t nb00 = src0->nb[0]; \ const uint32_t nb01 = src0->nb[1]; \ const uint32_t nb02 = src0->nb[2]; \ const uint32_t nb03 = src0->nb[3]; \ \ const uint32_t ne0 = dst->ne[0]; \ const uint32_t ne1 = dst->ne[1]; \ const uint32_t ne2 = dst->ne[2]; \ const uint32_t ne3 = dst->ne[3]; \ \ const uint32_t nb0 = dst->nb[0]; \ const uint32_t nb1 = dst->nb[1]; \ const uint32_t nb2 = dst->nb[2]; \ const uint32_t nb3 = dst->nb[3]; #define DEFINE_CPY_SAMESHAPE(NAME, ELEM_TYPE, ELEM_SIZE) \ static void cpy_thread_##NAME##_sameshape(unsigned int nth, unsigned int ith, void * data) { \ struct htp_copy_context * ct = (struct htp_copy_context *) data; \ struct htp_ops_context * octx = ct->octx; \ cpy_preamble; \ const uint32_t dr = ct->src0_nrows_per_thread; \ const uint32_t ir0 = ct->row_start + dr * ith; \ const uint32_t ir1 = MIN(ir0 + dr, ct->row_start + ct->nrows); \ if (ir0 >= ir1) return; \ const bool contiguous = htp_tensor_is_contiguous(src0, ELEM_SIZE) && htp_tensor_is_contiguous(dst, ELEM_SIZE); \ if (contiguous) { \ dma_queue * dma_q = octx->ctx->dma[ith]; \ dma_addr_t dst_addr = dst->data + ir0 * ne00 * ELEM_SIZE; \ dma_addr_t src0_addr = src0->data + ir0 * ne00 * ELEM_SIZE; \ cpy_dma_sametype_reshape_contig(dma_q, dst_addr, src0_addr, (ir1 - ir0) * ne00 * ELEM_SIZE); \ dma_queue_flush(dma_q); \ return; \ } \ const uint32_t ne02_ne01 = ne02 * ne01; \ uint32_t i03 = fastdiv(ir0, &ct->div_ne02_ne01); \ uint32_t rem = ir0 - i03 * ne02_ne01; \ uint32_t i02 = fastdiv(rem, &ct->div_ne01); \ uint32_t i01 = rem - i02 * ne01; \ uint8_t * dst_ptr = (uint8_t *) dst->data + i01*nb1 + i02*nb2 + i03*nb3; \ uint8_t * src0_ptr = (uint8_t *) src0->data + i01*nb01 + i02*nb02 + i03*nb03; \ for (uint32_t r = ir0; r < ir1; r++) { \ hex_l2fetch(src0_ptr, ne00 * ELEM_SIZE, nb01, 2); \ hvx_copy_uu(dst_ptr, src0_ptr, ne00, ELEM_SIZE); \ dst_ptr += nb1; \ src0_ptr += nb01; \ if (++i01 == ne01) { \ i01 = 0; \ if (++i02 == ne02) { \ i02 = 0; \ i03++; \ } \ dst_ptr = (uint8_t *) dst->data + i02*nb2 + i03*nb3; \ src0_ptr = (uint8_t *) src0->data + i02*nb02 + i03*nb03; \ } \ } \ } DEFINE_CPY_SAMESHAPE(f32, float, 4) DEFINE_CPY_SAMESHAPE(f16, __fp16, 2) DEFINE_CPY_SAMESHAPE(i32, int32_t, 4) #define DEFINE_CPY_RESHAPE(NAME, ELEM_TYPE, ELEM_SIZE) \ static void cpy_thread_##NAME##_reshape(unsigned int nth, unsigned int ith, void * data) { \ struct htp_copy_context * ct = (struct htp_copy_context *) data; \ struct htp_ops_context * octx = ct->octx; \ cpy_preamble; \ const uint32_t th_nelem = ct->elem_per_thread; \ const uint32_t th_start = ct->elem_start + ith * th_nelem; \ const uint32_t th_end = MIN(th_start + th_nelem, ct->elem_start + ct->nelem); \ if (th_start >= th_end) return; \ \ if (htp_tensor_is_contiguous(src0, ELEM_SIZE) && htp_tensor_is_contiguous(dst, ELEM_SIZE)) { \ dma_queue * dma_q = octx->ctx->dma[ith]; \ dma_addr_t dst_addr = dst->data + th_start * ELEM_SIZE; \ dma_addr_t src0_addr = src0->data + th_start * ELEM_SIZE; \ cpy_dma_sametype_reshape_contig(dma_q, dst_addr, src0_addr, (th_end - th_start) * ELEM_SIZE); \ dma_queue_flush(dma_q); \ return; \ } \ \ const uint32_t ne01_ne00 = ne01 * ne00; \ const uint32_t ne02_ne01_ne00 = ne02 * ne01_ne00; \ const uint32_t ne1_ne0 = ne1 * ne0; \ const uint32_t ne2_ne1_ne0 = ne2 * ne1_ne0; \ \ uint32_t e = th_start; \ uint32_t i13 = fastdiv(e, &ct->div_ne2_ne1_ne0); \ uint32_t rem = e - i13 * ne2_ne1_ne0; \ uint32_t i12 = fastdiv(rem, &ct->div_ne1_ne0); \ uint32_t rem2 = rem - i12 * ne1_ne0; \ uint32_t i11 = fastdiv(rem2, &ct->div_ne0); \ uint32_t i10 = rem2 - i11 * ne0; \ \ uint32_t i03 = fastdiv(e, &ct->div_ne02_ne01_ne00); \ uint32_t rem_s = e - i03 * ne02_ne01_ne00; \ uint32_t i02 = fastdiv(rem_s, &ct->div_ne01_ne00); \ uint32_t rem2_s = rem_s - i02 * ne01_ne00; \ uint32_t i01 = fastdiv(rem2_s, &ct->div_ne00); \ uint32_t i00 = rem2_s - i01 * ne00; \ \ char * dst_ptr = (char *) dst->data + i10*nb0 + i11*nb1 + i12*nb2 + i13*nb3; \ const char * src0_ptr = (const char *) src0->data + i00*nb00 + i01*nb01 + i02*nb02 + i03*nb03; \ \ const bool rows_contig = (nb00 == ELEM_SIZE) && (nb0 == ELEM_SIZE); \ \ while (e < th_end) { \ uint32_t run = 1; \ if (rows_contig) { \ run = MIN(MIN(ne00 - i00, ne0 - i10), th_end - e); \ hvx_copy_uu((uint8_t *) dst_ptr, (const uint8_t *) src0_ptr, run, ELEM_SIZE); \ } else { \ *((ELEM_TYPE *) dst_ptr) = *((const ELEM_TYPE *) src0_ptr); \ } \ e += run; \ \ dst_ptr += run * nb0; \ i10 += run; \ if (i10 == ne0) { \ i10 = 0; \ if (++i11 == ne1) { \ i11 = 0; \ if (++i12 == ne2) { \ i12 = 0; \ i13++; \ } \ } \ dst_ptr = (char *) dst->data + i11*nb1 + i12*nb2 + i13*nb3; \ } \ \ src0_ptr += run * nb00; \ i00 += run; \ if (i00 == ne00) { \ i00 = 0; \ if (++i01 == ne01) { \ i01 = 0; \ if (++i02 == ne02) { \ i02 = 0; \ i03++; \ } \ } \ src0_ptr = (const char *) src0->data + i01*nb01 + i02*nb02 + i03*nb03; \ } \ } \ } DEFINE_CPY_RESHAPE(f32, float, 4) DEFINE_CPY_RESHAPE(f16, __fp16, 2) DEFINE_CPY_RESHAPE(i32, int32_t, 4) static void cpy_thread_f16_f32_sameshape(unsigned int nth, unsigned int ith, void * data) { struct htp_copy_context * ct = (struct htp_copy_context *) data; struct htp_ops_context * octx = ct->octx; cpy_preamble; const uint32_t dr = ct->src0_nrows_per_thread; const uint32_t ir0 = ct->row_start + dr * ith; const uint32_t ir1 = MIN(ir0 + dr, ct->row_start + ct->nrows); if (ir0 >= ir1) return; const uint32_t ne02_ne01 = ne02 * ne01; uint32_t i03 = fastdiv(ir0, &ct->div_ne02_ne01); uint32_t rem = ir0 - i03 * ne02_ne01; uint32_t i02 = fastdiv(rem, &ct->div_ne01); uint32_t i01 = rem - i02 * ne01; uint8_t* dst_ptr = (uint8_t*) dst->data + i01*nb1 + i02*nb2 + i03*nb3; uint8_t* src0_ptr = (uint8_t*) src0->data + i01*nb01 + i02*nb02 + i03*nb03; for (uint32_t r = ir0; r < ir1; r++) { hex_l2fetch(src0_ptr, ne00 * sizeof(float), nb01, 2); hvx_copy_f16_f32_uu(dst_ptr, src0_ptr, ne00); dst_ptr += nb1; src0_ptr += nb01; if (++i01 == ne01) { i01 = 0; if (++i02 == ne02) { i02 = 0; i03++; } dst_ptr = (uint8_t*) dst->data + i02*nb2 + i03*nb3; src0_ptr = (uint8_t*) src0->data + i02*nb02 + i03*nb03; } } } static void cpy_thread_f32_f16_sameshape(unsigned int nth, unsigned int ith, void * data) { struct htp_copy_context * ct = (struct htp_copy_context *) data; struct htp_ops_context * octx = ct->octx; cpy_preamble; const uint32_t dr = ct->src0_nrows_per_thread; const uint32_t ir0 = ct->row_start + dr * ith; const uint32_t ir1 = MIN(ir0 + dr, ct->row_start + ct->nrows); if (ir0 >= ir1) return; const uint32_t ne02_ne01 = ne02 * ne01; uint32_t i03 = fastdiv(ir0, &ct->div_ne02_ne01); uint32_t rem = ir0 - i03 * ne02_ne01; uint32_t i02 = fastdiv(rem, &ct->div_ne01); uint32_t i01 = rem - i02 * ne01; uint8_t* dst_ptr = (uint8_t*) dst->data + i01*nb1 + i02*nb2 + i03*nb3; uint8_t* src0_ptr = (uint8_t*) src0->data + i01*nb01 + i02*nb02 + i03*nb03; for (uint32_t r = ir0; r < ir1; r++) { hex_l2fetch(src0_ptr, ne00 * sizeof(__fp16), nb01, 2); hvx_copy_f32_f16_uu(dst_ptr, src0_ptr, ne00); dst_ptr += nb1; src0_ptr += nb01; if (++i01 == ne01) { i01 = 0; if (++i02 == ne02) { i02 = 0; i03++; } dst_ptr = (uint8_t*) dst->data + i02*nb2 + i03*nb3; src0_ptr = (uint8_t*) src0->data + i02*nb02 + i03*nb03; } } } static void cpy_thread_i32_f32_sameshape(unsigned int nth, unsigned int ith, void * data) { struct htp_copy_context * ct = (struct htp_copy_context *) data; struct htp_ops_context * octx = ct->octx; cpy_preamble; const uint32_t dr = ct->src0_nrows_per_thread; const uint32_t ir0 = ct->row_start + dr * ith; const uint32_t ir1 = MIN(ir0 + dr, ct->row_start + ct->nrows); if (ir0 >= ir1) return; const uint32_t ne02_ne01 = ne02 * ne01; uint32_t i03 = fastdiv(ir0, &ct->div_ne02_ne01); uint32_t rem = ir0 - i03 * ne02_ne01; uint32_t i02 = fastdiv(rem, &ct->div_ne01); uint32_t i01 = rem - i02 * ne01; uint8_t* dst_ptr = (uint8_t*) dst->data + i01*nb1 + i02*nb2 + i03*nb3; uint8_t* src0_ptr = (uint8_t*) src0->data + i01*nb01 + i02*nb02 + i03*nb03; for (uint32_t r = ir0; r < ir1; r++) { hex_l2fetch(src0_ptr, ne00 * sizeof(float), nb01, 2); const float * restrict src_row = (const float *) src0_ptr; int32_t * restrict dst_row = (int32_t *) dst_ptr; for (uint32_t i = 0; i < ne00; i++) { dst_row[i] = (int32_t) src_row[i]; } dst_ptr += nb1; src0_ptr += nb01; if (++i01 == ne01) { i01 = 0; if (++i02 == ne02) { i02 = 0; i03++; } dst_ptr = (uint8_t*) dst->data + i02*nb2 + i03*nb3; src0_ptr = (uint8_t*) src0->data + i02*nb02 + i03*nb03; } } } static void cpy_thread_f32_i32_sameshape(unsigned int nth, unsigned int ith, void * data) { struct htp_copy_context * ct = (struct htp_copy_context *) data; struct htp_ops_context * octx = ct->octx; cpy_preamble; const uint32_t dr = ct->src0_nrows_per_thread; const uint32_t ir0 = ct->row_start + dr * ith; const uint32_t ir1 = MIN(ir0 + dr, ct->row_start + ct->nrows); if (ir0 >= ir1) return; const uint32_t ne02_ne01 = ne02 * ne01; uint32_t i03 = fastdiv(ir0, &ct->div_ne02_ne01); uint32_t rem = ir0 - i03 * ne02_ne01; uint32_t i02 = fastdiv(rem, &ct->div_ne01); uint32_t i01 = rem - i02 * ne01; uint8_t* dst_ptr = (uint8_t*) dst->data + i01*nb1 + i02*nb2 + i03*nb3; uint8_t* src0_ptr = (uint8_t*) src0->data + i01*nb01 + i02*nb02 + i03*nb03; for (uint32_t r = ir0; r < ir1; r++) { hex_l2fetch(src0_ptr, ne00 * sizeof(int32_t), nb01, 2); const int32_t * restrict src_row = (const int32_t *) src0_ptr; float * restrict dst_row = (float *) dst_ptr; for (uint32_t i = 0; i < ne00; i++) { dst_row[i] = (float) src_row[i]; } dst_ptr += nb1; src0_ptr += nb01; if (++i01 == ne01) { i01 = 0; if (++i02 == ne02) { i02 = 0; i03++; } dst_ptr = (uint8_t*) dst->data + i02*nb2 + i03*nb3; src0_ptr = (uint8_t*) src0->data + i02*nb02 + i03*nb03; } } } static int exec_cpy(struct htp_ops_context * octx, bool * use_dma) { cpy_preamble; *use_dma = false; const uint32_t total_elems_src = ne00 * ne01 * ne02 * ne03; const uint32_t total_elems_dst = ne0 * ne1 * ne2 * ne3; if (total_elems_src == 1 && total_elems_dst == 1) { if (octx->ctx->mdev.count > 1 && octx->ctx->mdev.idx > 0) { return HTP_STATUS_OK; } if (src0->type == HTP_TYPE_F32 && dst->type == HTP_TYPE_I32) { ((int32_t *) dst->data)[0] = (int32_t) (((const float *) src0->data)[0]); return HTP_STATUS_OK; } if (src0->type == HTP_TYPE_I32 && dst->type == HTP_TYPE_F32) { ((float *) dst->data)[0] = (float) (((const int32_t *) src0->data)[0]); return HTP_STATUS_OK; } if (src0->type == HTP_TYPE_I32 && dst->type == HTP_TYPE_I32) { ((int32_t *) dst->data)[0] = ((const int32_t *) src0->data)[0]; return HTP_STATUS_OK; } if (src0->type == HTP_TYPE_F32 && dst->type == HTP_TYPE_F32) { ((float *) dst->data)[0] = ((const float *) src0->data)[0]; return HTP_STATUS_OK; } if (src0->type == HTP_TYPE_F16 && dst->type == HTP_TYPE_F16) { ((__fp16 *) dst->data)[0] = ((const __fp16 *) src0->data)[0]; return HTP_STATUS_OK; } if (src0->type == HTP_TYPE_F32 && dst->type == HTP_TYPE_F16) { ((__fp16 *) dst->data)[0] = (__fp16) (((const float *) src0->data)[0]); return HTP_STATUS_OK; } if (src0->type == HTP_TYPE_F16 && dst->type == HTP_TYPE_F32) { ((float *) dst->data)[0] = (float) (((const __fp16 *) src0->data)[0]); return HTP_STATUS_OK; } } struct htp_copy_context ct; ct.octx = octx; switch (src0->type) { case HTP_TYPE_F32: ct.src0_type_size = 4; ct.src0_block_size = 1; ct.src0_blocks_per_row = ne00 / 1; break; case HTP_TYPE_F16: ct.src0_type_size = 2; ct.src0_block_size = 1; ct.src0_blocks_per_row = ne00 / 1; break; case HTP_TYPE_I32: ct.src0_type_size = 4; ct.src0_block_size = 1; ct.src0_blocks_per_row = ne00 / 1; break; default: return HTP_STATUS_NO_SUPPORT; } switch (dst->type) { case HTP_TYPE_F32: ct.dst_type_size = 4; ct.dst_block_size = 1; ct.dst_blocks_per_row = ne0 / 1; break; case HTP_TYPE_F16: ct.dst_type_size = 2; ct.dst_block_size = 1; ct.dst_blocks_per_row = ne0 / 1; break; case HTP_TYPE_I32: ct.dst_type_size = 4; ct.dst_block_size = 1; ct.dst_blocks_per_row = ne0 / 1; break; default: return HTP_STATUS_NO_SUPPORT; } const bool sametype = (src0->type == dst->type); const bool transposed = (nb00 > nb01) || (nb0 > nb1) || (nb00 != ct.src0_type_size) || (nb0 != ct.dst_type_size) || (nb01 < ne00 * ct.src0_type_size) || (nb1 < ne0 * ct.dst_type_size); const bool sameshape = !transposed && (ne00 == ne0 && ne01 == ne1 && ne02 == ne2 && ne03 == ne3); const uint32_t n_threads = octx->n_threads; const bool src_is_contiguous = htp_tensor_is_contiguous(src0, ct.src0_type_size); const bool dst_is_contiguous = htp_tensor_is_contiguous(dst, ct.dst_type_size); if (htp_tensor_is_extended(src0) || htp_tensor_is_extended(dst)) { if (!sametype) { return HTP_STATUS_NO_SUPPORT; } if (!sameshape && !(src_is_contiguous && dst_is_contiguous && octx->ctx->mdev.count <= 1)) { return HTP_STATUS_NO_SUPPORT; } } if (sameshape) { const uint32_t total_rows = ne01 * ne02 * ne03; const uint32_t row_size = ne00 * ct.dst_type_size; ct.div_ne01 = init_fastdiv_values(ne01); ct.div_ne02_ne01 = init_fastdiv_values(ne02 * ne01); uint32_t row_start = 0; uint32_t nrows = total_rows; if (octx->ctx->mdev.count > 1) { const uint32_t rows_per_chunk = (row_size > 0) ? (HEX_L2_LINE_SIZE / hex_gcd_u32(row_size, HEX_L2_LINE_SIZE)) : 1; const bool can_split = htp_tensor_mdev_data_aligned(dst) && dst_is_contiguous; const struct htp_tensor_mdev_range range = htp_tensor_mdev_partition(total_rows, can_split ? rows_per_chunk : 0, octx->ctx->mdev.idx, octx->ctx->mdev.count, &octx->ctx->mdev.count_div); row_start = range.start; nrows = range.count; } if (nrows == 0) { return HTP_STATUS_OK; } ct.row_start = row_start; ct.nrows = nrows; ct.src0_nrows_per_thread = fastdiv(nrows + n_threads - 1, &octx->n_threads_div); if (sametype && (octx->ctx->mdev.count <= 1 || htp_tensor_is_extended(src0) || htp_tensor_is_extended(dst))) { if (octx->ctx->mdev.idx == 0) { *use_dma = true; cpy_dma_sametype_sameshape(octx->ctx->dma[0], dst, src0, ct.src0_type_size); dma_queue_flush(octx->ctx->dma[0]); } } else { work_queue_func_t copy_fun = NULL; if (sametype) { switch (src0->type) { case HTP_TYPE_F32: copy_fun = cpy_thread_f32_sameshape; break; case HTP_TYPE_F16: copy_fun = cpy_thread_f16_sameshape; break; case HTP_TYPE_I32: copy_fun = cpy_thread_i32_sameshape; break; default: return HTP_STATUS_NO_SUPPORT; } } else if (dst->type == HTP_TYPE_F16 && src0->type == HTP_TYPE_F32) { copy_fun = cpy_thread_f16_f32_sameshape; } else if (dst->type == HTP_TYPE_F32 && src0->type == HTP_TYPE_F16) { copy_fun = cpy_thread_f32_f16_sameshape; } else if (dst->type == HTP_TYPE_I32 && src0->type == HTP_TYPE_F32) { copy_fun = cpy_thread_i32_f32_sameshape; } else if (dst->type == HTP_TYPE_F32 && src0->type == HTP_TYPE_I32) { copy_fun = cpy_thread_f32_i32_sameshape; } else { return HTP_STATUS_NO_SUPPORT; } work_queue_run(octx->ctx->work_queue, copy_fun, &ct, n_threads); } } else if (sametype) { const uint32_t total_elems = ne0 * ne1 * ne2 * ne3; const uint32_t elems_per_line = (ct.dst_type_size == 4) ? 32 : 64; if (octx->ctx->mdev.count <= 1 && dst_is_contiguous && src_is_contiguous) { *use_dma = true; cpy_dma_sametype_reshape_contig(octx->ctx->dma[0], dst->data, src0->data, total_elems * ct.dst_type_size); dma_queue_flush(octx->ctx->dma[0]); return HTP_STATUS_OK; } ct.div_ne0 = init_fastdiv_values(ne0); ct.div_ne1_ne0 = init_fastdiv_values(ne1 * ne0); ct.div_ne2_ne1_ne0 = init_fastdiv_values(ne2 * ne1 * ne0); ct.div_ne00 = init_fastdiv_values(ne00); ct.div_ne01_ne00 = init_fastdiv_values(ne01 * ne00); ct.div_ne02_ne01_ne00 = init_fastdiv_values(ne02 * ne01 * ne00); uint32_t elem_start = 0; uint32_t nelem = total_elems; if (octx->ctx->mdev.count > 1) { const bool can_split = htp_tensor_mdev_data_aligned(dst) && dst_is_contiguous; const struct htp_tensor_mdev_range range = htp_tensor_mdev_partition(total_elems, can_split ? elems_per_line : 0, octx->ctx->mdev.idx, octx->ctx->mdev.count, &octx->ctx->mdev.count_div); elem_start = range.start; nelem = range.count; } if (nelem == 0) { return HTP_STATUS_OK; } ct.elem_start = elem_start; ct.nelem = nelem; ct.elem_per_thread = fastdiv(nelem + n_threads - 1, &octx->n_threads_div); work_queue_func_t copy_fun = NULL; switch (src0->type) { case HTP_TYPE_F32: copy_fun = cpy_thread_f32_reshape; break; case HTP_TYPE_F16: copy_fun = cpy_thread_f16_reshape; break; case HTP_TYPE_I32: copy_fun = cpy_thread_i32_reshape; break; default: return HTP_STATUS_NO_SUPPORT; } work_queue_run(octx->ctx->work_queue, copy_fun, &ct, n_threads); } else { return HTP_STATUS_NO_SUPPORT; } return HTP_STATUS_OK; } int op_cpy(struct htp_ops_context * octx) { bool use_dma = false; int status = exec_cpy(octx, &use_dma); htp_ops_context_set_status(octx, status); if (octx->op == HTP_OP_CPY_FENCE) { if (!use_dma) { htp_flush_dirty_ranges(octx->ctx); } htp_mdev_group_barrier(octx); if (octx->ctx->mdev.idx == 0) { const struct htp_tensor * sync = octx->src[1]; if (htp_tensor_is_extended(sync)) { return HTP_STATUS_NO_SUPPORT; } const uint32_t seq = (uint32_t) octx->op_params[0]; atomic_uint * sync_fence = (atomic_uint *) (uintptr_t) sync->data; htp_fence_write(sync_fence, seq, octx->status); FARF(HIGH, "ggml-hex: sync-release : fence %p seq 0x%x status %d\n", sync_fence, seq, octx->status); } } return octx->status; }