yizhi-js-lib/ai-box
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

初次提交

Browse Source
This commit is contained in:
Yizhi
2025-03-06 14:38:32 +08:00
commit b0e71d0ebb
46 changed files with 1926 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
cxx/test.cc Normal file
View File

@ -0,0 +1,53 @@
#include <onnxruntime_cxx_api.h>
static void buildNames(const std::vector<std::string> &input, std::vector<const char *> &output)
{
if (input.size() == output.size())
return;
for (auto &it : input)
output.push_back(it.c_str());
}
void test()
{
static Ort::Env env;
static Ort::SessionOptions sessionOptions;
static Ort::Session *session_ = new Ort::Session(env, "/home/yizhi/Develop/ai-box/models/test.onnx", sessionOptions);
static std::vector<std::string> inputNames_;
static std::vector<const char *> inputNames_c_;
static std::vector<Ort::Value> inputValues_;
static std::vector<std::string> outputNames_;
static std::vector<const char *> outputNames_c_;
static std::vector<Ort::Value> outputValues_;
inputNames_.emplace_back("a");
inputNames_.emplace_back("b");
outputNames_.emplace_back("c");
outputValues_.emplace_back(Ort::Value{nullptr});
static std::vector<int64_t> shapeA = {3, 4};
static std::vector<int64_t> shapeB = {4, 3};
static float inputA[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12};
static float inputB[] = {10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120};
auto memoryInfo = Ort::MemoryInfo::CreateCpu(OrtDeviceAllocator, OrtMemTypeDefault);
inputValues_.emplace_back(Ort::Value::CreateTensor(memoryInfo, inputA, sizeof(inputA), &shapeA[0], shapeA.size(), ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT));
inputValues_.emplace_back(Ort::Value::CreateTensor(memoryInfo, inputB, sizeof(inputB), &shapeB[0], shapeB.size(), ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT));
buildNames(inputNames_, inputNames_c_);
buildNames(outputNames_, outputNames_c_);
const char **ptr = &inputNames_c_[0];
for (int i = 0; i < inputNames_c_.size(); ++i)
{
printf("input [%d] = %s\n", i, ptr[i]);
}
Ort::RunOptions runOption;
printf("start run\n");
session_->Run(runOption, &inputNames_c_[0], &inputValues_[0], inputNames_.size(), &outputNames_c_[0], &outputValues_[0], outputNames_.size());
printf("end run\n");
}
int main(int argc, char **argv)
{
Ort::Env env;
test();
// ai::ORTSession session(nullptr, 0);
return 0;
}