6. 使用 ONNXRuntime 进行预测¶
API定义如下:
// 启用 ONNXRuntime 进行预测
// 参数:None
// 返回:None
void PD_ConfigEnableONNXRuntime(PD_Config* pd_config);
// 禁用 ONNXRuntime 进行预测
// 参数:None
// 返回:None
void PD_ConfigDisableONNXRuntime(PD_Config* pd_config);
// 判断是否启用 ONNXRuntime
// 参数:None
// 返回:bool - 是否启用 ONNXRuntime
PD_Bool PD_ConfigONNXRuntimeEnabled(PD_Config* pd_config);
// 启用 ONNXRuntime 预测时开启优化
// 参数:None
// 返回:None
void PD_ConfigEnableORTOptimization(PD_Config* pd_config);
ONNXRuntime设置代码示例:
// 创建 Config 对象
PD_Config* config = PD_ConfigCreate();
// 启用 ONNXRuntime
PD_ConfigEnableONNXRuntime(config);
// 通过 API 获取 ONNXRuntime 信息
printf("Use ONNXRuntime is: %s\n", PD_ConfigONNXRuntimeEnabled(config) ? "True" : "False"); // True
// 开启ONNXRuntime优化
PD_ConfigEnableORTOptimization(config);
// 设置 ONNXRuntime 算子计算线程数为 10
PD_ConfigSetCpuMathLibraryNumThreads(config, 10);
// 禁用 ONNXRuntime 进行预测
PD_ConfigDisableONNXRuntime(config);
// 通过 API 获取 ONNXRuntime 信息
printf("Use ONNXRuntime is: %s\n", PD_ConfigONNXRuntimeEnabled(config) ? "True" : "False"); // False