AscendIndexIVFSQTConfig
AscendIndexIVFSQT需要使用对应的AscendIndexIVFSQTConfig执行对应资源的初始化。
AscendIndexIVFSQTConfig
API定义 |
inline AscendIndexIVFSQTConfig(); |
---|---|
功能描述 |
默认构造函数,默认devices为{0},使用第0个昇腾AI处理器进行计算,默认resource为384MB。 |
输入 |
无 |
输出 |
无 |
返回值 |
无 |
约束说明 |
无 |
API定义 |
inline AscendIndexIVFSQTConfig(std::initializer_list<int> devices, int64_t resourceSize = IVFSQT_DEFAULT_TEMP_MEM); |
---|---|
功能描述 |
AscendIndexIVFSQTConfig的构造函数,生成AscendIndexIVFSQTConfig,此时根据“devices”中配置的值设置Device侧昇腾AI处理器资源,配置资源池大小并执行默认的初始化。 |
输入 |
std::initializer_list<int> devices:Device侧设备ID。 int64_t resourceSize:设备侧预置的内存池大小,单位为Byte,计算过程中存储中间结果的内存空间,用于避免计算过程中动态申请内存造成性能波动。默认参数为头文件中的“IVFSQT_DEFAULT_TEMP_MEM”。该参数通过底库大小和search的batch数共同确定,在底库大于或等于1000万且batch数大于或等于16时建议设置1024MB。 |
输出 |
无 |
返回值 |
无 |
约束说明 |
|
API定义 |
inline AscendIndexIVFSQTConfig(std::vector<int> devices, int64_t resourceSize = IVFSQT_DEFAULT_TEMP_MEM); |
---|---|
功能描述 |
AscendIndexIVFSQTConfig的构造函数,生成AscendIndexIVFSQTConfig,此时根据“devices”中配置的值设置Device侧昇腾AI处理器资源,配置资源池大小并执行默认的初始化。 |
输入 |
std::vector<int> devices:Device侧设备ID。 int64_t resourceSize:设备侧预置的内存池大小,单位为Byte,计算过程中存储中间结果的内存空间,用于避免计算过程中动态申请内存造成性能波动。默认参数为头文件中的“IVFSQT_DEFAULT_TEMP_MEM”。该参数通过底库大小和search的batch数共同确定,在底库大于或等于1000万且batch数大于或等于16时建议设置1024MB。 |
输出 |
无 |
返回值 |
无 |
约束说明 |
|
SetDefaultIVFSQConfig
API定义 |
inline void SetDefaultIVFSQConfig(); |
---|---|
功能描述 |
执行默认的初始化,设置迭代数为16,每个centroids最多设置512个点。 |
输入 |
无 |
输出 |
无 |
返回值 |
无 |
约束说明 |
无 |