forked from ccf-ai-infra/Intro-ops
34 lines
756 B
C
34 lines
756 B
C
#pragma once
|
|
|
|
#include "operator_runtime/api.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
OPRT_EXPORT oprt_status_t oprt_create_vector_add_descriptor(
|
|
oprt_operator_descriptor_t *desc,
|
|
const oprt_tensor_view_t *out,
|
|
const oprt_tensor_view_t *a,
|
|
const oprt_tensor_view_t *b);
|
|
|
|
OPRT_EXPORT oprt_status_t oprt_get_vector_add_workspace_size(
|
|
oprt_operator_descriptor_t desc,
|
|
size_t *size);
|
|
|
|
OPRT_EXPORT oprt_status_t oprt_execute_vector_add(
|
|
oprt_operator_descriptor_t desc,
|
|
void *workspace,
|
|
size_t workspace_size,
|
|
void *out,
|
|
const void *a,
|
|
const void *b,
|
|
oprt_stream_t stream);
|
|
|
|
OPRT_EXPORT oprt_status_t oprt_destroy_vector_add_descriptor(
|
|
oprt_operator_descriptor_t desc);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|