e25f20f7a3
commit 3f4938648950a7f3bf9a19c320ca9fae7c52de20 Author: sophgo-forum-service <forum_service@sophgo.com> Date: Mon May 13 13:44:23 2024 +0800 [feat] cviruntime opensource for cv18xx soc. - a4b6a3, add cumsum and gatherelements_pt.
34 lines
898 B
C++
34 lines
898 B
C++
#ifndef RUNTIME_TDMA_COPY_HPP
|
|
#define RUNTIME_TDMA_COPY_HPP
|
|
|
|
#include "cviruntime_context.h"
|
|
#include "cviruntime.h"
|
|
#include "cvikernel/cvikernel.h"
|
|
|
|
namespace cvi {
|
|
namespace runtime {
|
|
|
|
CVI_RC runtimeExecuteKernelFunction(
|
|
CVI_RT_HANDLE ctx, CVI_RT_MEM codeBuf,
|
|
uint64_t gaddrSrc, uint64_t gaddrDst);
|
|
|
|
CVI_RT_MEM runtimeJitTdmaStrideCopy(
|
|
CVI_RT_HANDLE ctx, void *cvk, CVI_FMT fmt,
|
|
cvk_tg_shape_t *shapeDst, cvk_tg_stride_t *strideDst,
|
|
cvk_tg_shape_t *shapeSrc, cvk_tg_stride_t *strideSrc);
|
|
|
|
CVI_RT_MEM runtimeJitMatrixMul(
|
|
CVI_RT_HANDLE ctx, void* cvk_ctx, CVI_FMT fmt,
|
|
uint32_t m, uint32_t k, uint32_t n);
|
|
|
|
CVI_RT_MEM runtimeJitEuclideanDistance(
|
|
CVI_RT_HANDLE ctx, void* cvk_ctx,
|
|
uint32_t records, uint32_t feature_size);
|
|
|
|
CVI_RT_MEM runtimeJitGrayImageLight(
|
|
CVI_RT_HANDLE ctx, void* cvk_ctx,
|
|
int32_t ih, int32_t iw, int32_t kernel_sz);
|
|
}
|
|
|
|
}
|
|
#endif |