rte_mldev_pmd.h

Go to the documentation of this file.

/* SPDX-License-Identifier: BSD-3-Clause
 * Copyright (c) 2022 Marvell.
 */

#ifndef RTE_MLDEV_PMD_H
#define RTE_MLDEV_PMD_H

#ifdef __cplusplus
extern "C" {
#endif

#include <stdint.h>

#include <rte_common.h>
#include <rte_compat.h>
#include <rte_mldev.h>
#include <rte_mldev_core.h>

struct rte_ml_dev_pmd_init_params {
    uint8_t socket_id;

    uint64_t private_data_size;
};

struct rte_ml_dev;

__rte_internal
struct rte_ml_dev *
rte_ml_dev_pmd_get_dev(int16_t dev_id);

__rte_internal
struct rte_ml_dev *
rte_ml_dev_pmd_get_named_dev(const char *name);

__rte_internal
struct rte_ml_dev *
rte_ml_dev_pmd_allocate(const char *name, uint8_t socket_id);

__rte_internal
int
rte_ml_dev_pmd_release(struct rte_ml_dev *dev);

__rte_internal
struct rte_ml_dev *
rte_ml_dev_pmd_create(const char *name, struct rte_device *device,
              struct rte_ml_dev_pmd_init_params *params);

__rte_internal
int
rte_ml_dev_pmd_destroy(struct rte_ml_dev *mldev);

#ifdef __cplusplus
}
#endif

#endif /* RTE_MLDEV_PMD_H */