mirror of https://github.com/F-Stack/f-stack.git
41 lines
879 B
C
41 lines
879 B
C
/* SPDX-License-Identifier: BSD-3-Clause
|
|
* Copyright (c) 2023 Marvell.
|
|
*/
|
|
|
|
#ifndef _MVTVM_ML_DEV_H_
|
|
#define _MVTVM_ML_DEV_H_
|
|
|
|
#include <rte_mldev_core.h>
|
|
|
|
/* Device status */
|
|
extern int cnxk_ml_dev_initialized;
|
|
|
|
/* CNXK Device ops */
|
|
extern struct rte_ml_dev_ops cnxk_ml_ops;
|
|
|
|
/* Marvell MVTVM ML PMD device name */
|
|
#define MLDEV_NAME_MVTVM_PMD ml_mvtvm
|
|
|
|
/* Maximum number of descriptors per queue-pair */
|
|
#define ML_MVTVM_MAX_DESC_PER_QP 1024
|
|
|
|
/* Maximum number of inputs / outputs per model */
|
|
#define ML_MVTVM_MAX_INPUT_OUTPUT 32
|
|
|
|
/* Maximum number of segments for IO data */
|
|
#define ML_MVTVM_MAX_SEGMENTS 1
|
|
|
|
/* Device private data */
|
|
struct mvtvm_ml_dev {
|
|
/* Virtual device */
|
|
struct rte_vdev_device *vdev;
|
|
|
|
/* Maximum number of queue pairs */
|
|
uint16_t max_nb_qpairs;
|
|
|
|
/* Enable / disable model data caching */
|
|
int cache_model_data;
|
|
};
|
|
|
|
#endif /* _MVTVM_ML_DEV_H_ */
|