mirror of https://github.com/F-Stack/f-stack.git
88 lines
1.9 KiB
C
88 lines
1.9 KiB
C
|
/* SPDX-License-Identifier: BSD-3-Clause
|
||
|
|
* Copyright (c) 2023 Marvell.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifndef _CNXK_ML_OPS_H_
|
||
|
|
#define _CNXK_ML_OPS_H_
|
||
|
|
|
||
|
|
#include <rte_mldev.h>
|
||
|
|
#include <rte_mldev_core.h>
|
||
|
|
|
||
|
|
#include <roc_api.h>
|
||
|
|
|
||
|
|
#include "cn10k_ml_ops.h"
|
||
|
|
|
||
|
|
#ifdef RTE_MLDEV_CNXK_ENABLE_MVTVM
|
||
|
|
#include "mvtvm_ml_ops.h"
|
||
|
|
#else
|
||
|
|
#include "mvtvm_ml_stubs.h"
|
||
|
|
#endif
|
||
|
|
|
||
|
|
/* Request structure */
|
||
|
|
struct cnxk_ml_req {
|
||
|
|
/* Device specific request */
|
||
|
|
union {
|
||
|
|
/* CN10K */
|
||
|
|
struct cn10k_ml_req cn10k_req;
|
||
|
|
|
||
|
|
#ifdef RTE_MLDEV_CNXK_ENABLE_MVTVM
|
||
|
|
/* MVTVM */
|
||
|
|
struct mvtvm_ml_req mvtvm_req;
|
||
|
|
#endif
|
||
|
|
};
|
||
|
|
|
||
|
|
/* Address of status field */
|
||
|
|
volatile uint64_t *status;
|
||
|
|
|
||
|
|
/* Timeout cycle */
|
||
|
|
uint64_t timeout;
|
||
|
|
|
||
|
|
/* Op */
|
||
|
|
struct rte_ml_op *op;
|
||
|
|
} __rte_aligned(ROC_ALIGN);
|
||
|
|
|
||
|
|
/* Request queue */
|
||
|
|
struct cnxk_ml_queue {
|
||
|
|
/* Array of requests */
|
||
|
|
struct cnxk_ml_req *reqs;
|
||
|
|
|
||
|
|
/* Head of the queue, used for enqueue */
|
||
|
|
uint64_t head;
|
||
|
|
|
||
|
|
/* Tail of the queue, used for dequeue */
|
||
|
|
uint64_t tail;
|
||
|
|
|
||
|
|
/* Wait cycles before timeout */
|
||
|
|
uint64_t wait_cycles;
|
||
|
|
};
|
||
|
|
|
||
|
|
/* Queue-pair structure */
|
||
|
|
struct cnxk_ml_qp {
|
||
|
|
/* ID */
|
||
|
|
uint32_t id;
|
||
|
|
|
||
|
|
/* Number of descriptors */
|
||
|
|
uint64_t nb_desc;
|
||
|
|
|
||
|
|
/* Request queue */
|
||
|
|
struct cnxk_ml_queue queue;
|
||
|
|
|
||
|
|
/* Statistics per queue-pair */
|
||
|
|
struct rte_ml_dev_stats stats;
|
||
|
|
};
|
||
|
|
|
||
|
|
extern struct rte_ml_dev_ops cnxk_ml_ops;
|
||
|
|
|
||
|
|
int cnxk_ml_model_unload(struct rte_ml_dev *dev, uint16_t model_id);
|
||
|
|
int cnxk_ml_model_stop(struct rte_ml_dev *dev, uint16_t model_id);
|
||
|
|
void cnxk_ml_xstats_model_name_update(struct cnxk_ml_dev *cnxk_mldev, uint16_t model_id);
|
||
|
|
|
||
|
|
__rte_hot uint16_t cnxk_ml_enqueue_burst(struct rte_ml_dev *dev, uint16_t qp_id,
|
||
|
|
struct rte_ml_op **ops, uint16_t nb_ops);
|
||
|
|
__rte_hot uint16_t cnxk_ml_dequeue_burst(struct rte_ml_dev *dev, uint16_t qp_id,
|
||
|
|
struct rte_ml_op **ops, uint16_t nb_ops);
|
||
|
|
__rte_hot void cnxk_ml_set_poll_ptr(struct cnxk_ml_req *req);
|
||
|
|
__rte_hot uint64_t cnxk_ml_get_poll_ptr(struct cnxk_ml_req *req);
|
||
|
|
|
||
|
|
#endif /* _CNXK_ML_OPS_H_ */
|