下载
中文
注册

Init

API定义

virtual APP_ERROR Init(int dim, int capacity, AscendMetricType metricType, int64_t resourceSize) = 0;

功能描述

初始化特征库参数,申请底库内存资源。

输入

int dim:特征向量的维度。

AscendMetricType metricType: 特征距离类别:向量内积、欧氏距离、余弦相似度。

int capacity:底库最大容量,等于 cap * dim * sizeof(float) 字节内存数据。

int resourceSize:提前申请Device的缓存资源,检索接口被调用时可以直接使用这里的资源,而不必调用aclrtmalloc去申请内存,达到优化加速。默认取值-1,代表按默认size申请缓存资源(128MB),可以根据检索业务的数据量和Device上的资源使用情况来更精确地配置实际需要使用的size大小。

例如:query的“batch”“64”,底库总量为100万,而一个fp32数值占用4个字节,那么这里的“resourceSize”可以设置为: 64 * 1000000 * 4 = 256,000,000Byte。

输出

返回值

APP_ERROR:调用返回状态,具体请参见接口调用返回值参考

约束说明

入参由该类的实现类进行约束。