! * \brief Main logics of library loading and function calling. * \param x The input tensor. * \param y The output tensor. */
| 29 | * \param y The output tensor. |
| 30 | */ |
| 31 | void Run(tvm::ffi::TensorView x, tvm::ffi::TensorView y) { |
| 32 | // Load shared library `build/add_one_cpu.so` |
| 33 | ffi::Module mod = ffi::Module::LoadFromFile("build/add_one_cpu.so"); |
| 34 | // Look up `add_one_cpu` function |
| 35 | ffi::Function add_one_cpu = mod->GetFunction("add_one_cpu").value(); |
| 36 | // Call the function |
| 37 | add_one_cpu(x, y); |
| 38 | } |
| 39 | } // namespace |
| 40 | // [main.end] |
| 41 | /************* Auxiliary Logics *************/ |
no test coverage detected