/*
  Copyright (c), 2001-2022, Shenshu Tech. Co., Ltd.
 */

#ifndef _SS_MPI_NNIE_H_
#define _SS_MPI_NNIE_H_

#include "ot_common_nnie.h"

#ifdef __cplusplus
extern "C" {
#endif

td_s32 ss_mpi_svp_nnie_load_model(const ot_svp_src_mem_info *model_buf, ot_svp_nnie_model *model);

td_s32 ss_mpi_svp_nnie_get_task_buf_size(td_u32 max_batch_num, td_u32 max_bbox_num,
    const ot_svp_nnie_model *model, ot_svp_nnie_task_buf_info task_buf_info[], td_u32 net_seg_num);

td_s32 ss_mpi_svp_nnie_forward(ot_svp_nnie_handle *handle,
    const ot_svp_src_blob src[], const ot_svp_nnie_model *model, const ot_svp_dst_blob dst[],
    const ot_svp_nnie_forward_ctrl *forward_ctrl, td_bool is_instant);

td_s32 ss_mpi_svp_nnie_unload_model(ot_svp_nnie_model *model);

td_s32 ss_mpi_svp_nnie_set_prior_box(const ot_svp_src_blob prior_box[],
    td_u32 prior_box_num, ot_svp_nnie_model *model);

td_s32 ss_mpi_svp_nnie_query(ot_svp_nnie_id nnie_id, ot_svp_nnie_handle handle,
    td_bool *is_finish, td_bool is_block);

td_s32 ss_mpi_svp_nnie_add_mem_info(const ot_svp_mem_info *mem_info);

td_s32 ss_mpi_svp_nnie_rm_mem_info(const ot_svp_mem_info *mem_info);

#ifdef __cplusplus
}
#endif

#endif