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:调用返回状态,具体请参见接口调用返回值参考。 |
约束说明 |
入参由该类的实现类进行约束。 |
父主题: IndexIL