|
|
#include "runner/ax650/ax_api_loader.h" |
|
|
#include "runner/ax650/ax_model_runner_ax650.hpp" |
|
|
|
|
|
#include <fstream> |
|
|
#include <vector> |
|
|
#include <cstring> |
|
|
|
|
|
AxSysApiLoader &get_ax_sys_loader(); |
|
|
|
|
|
AxEngineApiLoader &get_ax_engine_loader(); |
|
|
|
|
|
int main() |
|
|
{ |
|
|
AxSysApiLoader &ax_sys_loader = get_ax_sys_loader(); |
|
|
|
|
|
AxEngineApiLoader &ax_engine_loader = get_ax_engine_loader(); |
|
|
|
|
|
ax_sys_loader.AX_SYS_Init(); |
|
|
AX_ENGINE_NPU_ATTR_T npu_attr; |
|
|
memset(&npu_attr, 0, sizeof(AX_ENGINE_NPU_ATTR_T)); |
|
|
npu_attr.eHardMode = AX_ENGINE_VIRTUAL_NPU_DISABLE; |
|
|
ax_engine_loader.AX_ENGINE_Init(&npu_attr); |
|
|
|
|
|
ax_runner_ax650 runner; |
|
|
std::ifstream file("cnclip/cnclip_vit_l14_336px_text_u16.axmodel", 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); |
|
|
|
|
|
ax_engine_loader.AX_ENGINE_Deinit(); |
|
|
ax_sys_loader.AX_SYS_Deinit(); |
|
|
return 0; |
|
|
} |
