Function main

samples/cpp/api/api_intro.cc:95–154

}

int main(void) {
    const char* model_file = "tests/testdata/conv.onnx";

    auto x86_engine = x86::EngineFactory::Create(x86::EngineOptions());
    vector<unique_ptr<Engine>> engines;
    engines.emplace_back(unique_ptr<Engine>(x86_engine));

    auto builder = unique_ptr<onnx::RuntimeBuilder>(onnx::RuntimeBuilderFactory::Create());
    if (!builder) {
        cerr << "create RuntimeBuilder failed." << endl;
        return -1;
    }

    auto status = builder->LoadModel(model_file);
    if (status != RC_SUCCESS) {
        cerr << "init OnnxRuntimeBuilder failed: " << GetRetCodeStr(status) << endl;
        return -1;
    }

    vector<Engine*> engine_ptrs(engines.size());
    for (uint32_t i = 0; i < engines.size(); ++i) {
        engine_ptrs[i] = engines[i].get();
    }
    onnx::RuntimeBuilder::Resources resources;
    resources.engines = engine_ptrs.data();
    resources.engine_num = engine_ptrs.size();

    status = builder->SetResources(resources);
    if (status != RC_SUCCESS) {
        cerr << "onnx RuntimeBuilder SetResources failed: " << GetRetCodeStr(status) << endl;
        return -1;
    }

    status = builder->Preprocess();
    if (status != RC_SUCCESS) {
        cerr << "builder preprocess failed: " << GetRetCodeStr(status) << endl;
        return -1;
    }

    auto runtime = unique_ptr<Runtime>(builder->CreateRuntime());
    if (!runtime) {
        cerr << "CreateRuntime failed." << endl;
        return -1;
    }

    if (!SetRandomInputs(runtime.get())) {
        cerr << "SetRandomInputs failed." << endl;
        return -1;
    }

    status = runtime->Run();
    if (status != RC_SUCCESS) {
        cerr << "Run() failed: " << GetRetCodeStr(status) << endl;
        return -1;
    }

    PrintInputOutputInfo(runtime.get());