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.
75 lines
1.9 KiB
C++
75 lines
1.9 KiB
C++
/*
|
|
* Copyright (C) Cvitek Co., Ltd. 2019-2020. All rights reserved.
|
|
*/
|
|
|
|
#ifndef RUNTIME_CPU_FUNCTION_H
|
|
#define RUNTIME_CPU_FUNCTION_H
|
|
#include <vector>
|
|
#include <stdint.h>
|
|
#include <memory>
|
|
#include <runtime/neuron.hpp>
|
|
#include <runtime/op_param.hpp>
|
|
|
|
namespace cvi {
|
|
namespace runtime {
|
|
|
|
class ICpuFunction {
|
|
public:
|
|
ICpuFunction() {}
|
|
virtual ~ICpuFunction() {}
|
|
virtual void setup(tensor_list_t &inputs,
|
|
tensor_list_t &outputs,
|
|
OpParam ¶m) = 0;
|
|
virtual void run() = 0;
|
|
|
|
protected:
|
|
template <typename T>
|
|
void print_data(T data) {
|
|
if (sizeof(T) == 4) {
|
|
printf("%e ", (float)data);
|
|
} else if (sizeof(T) == 1) {
|
|
printf("%4d ", (int)data);
|
|
} else {
|
|
assert(0);
|
|
std::cout << data << " ";
|
|
}
|
|
}
|
|
|
|
template <typename T>
|
|
void dump(const std::string &tag, const T *data, size_t count) {
|
|
auto ptr = (T *)data;
|
|
int loop = count / 10;
|
|
std::cout << "-------Dump " << tag << ", size:" << count << "\n";
|
|
|
|
for (int i = 0; i < loop; i++) {
|
|
for (int j = 0; j < 10; j++) {
|
|
print_data(*(ptr++));
|
|
}
|
|
std::cout << "\n";
|
|
}
|
|
for (int j = 0; j < (int)(count % 10); j++) {
|
|
print_data(*(ptr++));
|
|
}
|
|
std::cout << "\n";
|
|
}
|
|
};
|
|
|
|
typedef ICpuFunction *(*ICpuFunctionCreate)();
|
|
|
|
} // namespace runtime
|
|
} // namespace cvi
|
|
|
|
typedef struct {
|
|
char *name;
|
|
cvi::runtime::ICpuFunctionCreate func;
|
|
} CustomOpRuntimeFunc;
|
|
|
|
#define REGISTER_OP_RUNTIME_FUNCS(X, ...) \
|
|
extern "C" { \
|
|
CustomOpRuntimeFunc customOpRuntimeFuncs[] = {X, ##__VA_ARGS__}; \
|
|
int customOpRuntimeFuncsNum = \
|
|
sizeof(customOpRuntimeFuncs) / sizeof(CustomOpRuntimeFunc); \
|
|
}
|
|
|
|
#endif
|