Function ml_request_finish

dpdk/app/test-mldev/test_inference_common.c:1005–1118

Callback for mempool object iteration. This call would dequantize output data. */

/* Callback for mempool object iteration. This call would dequantize output data. */
static void
ml_request_finish(struct rte_mempool *mp, void *opaque, void *obj, unsigned int obj_idx)
{
	struct test_inference *t = ml_test_priv((struct ml_test *)opaque);
	struct ml_request *req = (struct ml_request *)obj;
	struct ml_model *model = &t->model[req->fid];
	bool error = false;
	char *dump_path;

	struct rte_ml_buff_seg qbuff_seg[ML_TEST_MAX_IO_SIZE];
	struct rte_ml_buff_seg dbuff_seg[ML_TEST_MAX_IO_SIZE];
	struct rte_ml_buff_seg *q_segs[ML_TEST_MAX_IO_SIZE];
	struct rte_ml_buff_seg *d_segs[ML_TEST_MAX_IO_SIZE];
	uint64_t offset;
	uint64_t bufsz;
	uint32_t i;

	RTE_SET_USED(mp);

	if (req->niters == 0)
		return;

	t->nb_used++;

	if (t->model[req->fid].info.io_layout == RTE_ML_IO_LAYOUT_PACKED) {
		qbuff_seg[0].addr = req->output;
		qbuff_seg[0].iova_addr = rte_mem_virt2iova(req->output);
		qbuff_seg[0].length = t->model[req->fid].out_qsize;
		qbuff_seg[0].next = NULL;
		q_segs[0] = &qbuff_seg[0];

		dbuff_seg[0].addr = model->output;
		dbuff_seg[0].iova_addr = rte_mem_virt2iova(model->output);
		dbuff_seg[0].length = t->model[req->fid].out_dsize;
		dbuff_seg[0].next = NULL;
		d_segs[0] = &dbuff_seg[0];
	} else {
		offset = 0;
		for (i = 0; i < t->model[req->fid].info.nb_outputs; i++) {
			bufsz = RTE_ALIGN_CEIL(t->model[req->fid].info.output_info[i].size,
					       t->cmn.dev_info.align_size);
			qbuff_seg[i].addr = req->output + offset;
			qbuff_seg[i].iova_addr = rte_mem_virt2iova(req->output + offset);
			qbuff_seg[i].length = bufsz;
			qbuff_seg[i].next = NULL;
			q_segs[i] = &qbuff_seg[i];
			offset += bufsz;
		}

		offset = 0;
		for (i = 0; i < t->model[req->fid].info.nb_outputs; i++) {
			bufsz = t->model[req->fid].info.output_info[i].nb_elements * sizeof(float);
			dbuff_seg[i].addr = model->output + offset;
			dbuff_seg[i].iova_addr = rte_mem_virt2iova(model->output + offset);
			dbuff_seg[i].length = bufsz;
			dbuff_seg[i].next = NULL;
			d_segs[i] = &dbuff_seg[i];
			offset += bufsz;