| #include "runner/axcl/axcl_manager.h" |
| #include "runner/axcl/ax_model_runner_axcl.hpp" |
| #include "cmdline.hpp" |
| #include <fstream> |
|
|
| int main(int argc, char *argv[]) |
| { |
| auto ret = axclInit(); |
| if (ret != 0) |
| { |
| printf("axclInit failed\n"); |
| return -1; |
| } |
|
|
| axcl_Dev_Init(0); |
| cmdline::parser parser; |
| parser.add<std::string>("model", 'm', "model", true); |
| parser.parse_check(argc, argv); |
|
|
| ax_runner_axcl runner; |
| std::ifstream file(parser.get<std::string>("model"), std::ios::binary); |
| if (!file.is_open()) |
| { |
| printf("open file failed\n"); |
| return -1; |
| } |
| std::vector<uint8_t> model_data((std::istreambuf_iterator<char>(file)), std::istreambuf_iterator<char>()); |
| runner.init(model_data.data(), model_data.size(), 0); |
|
|
| runner.deinit(); |
|
|
| axcl_Dev_Exit(0); |
| axclFinalize(); |
| return 0; |
| } |