鲲鹏BoostKit搜推广使能套件旨在为互联网搜索、推荐、广告业务场景提供基于鲲鹏平台的应用层加速能力,组件涵盖召回场景核心检索算法、排序场景模型推理软件框架优化。
| 算法名称 | 描述 | 仓库路径 |
|---|---|---|
| KBest召回算法 | Kunpeng Blazing-fast embedding similarity search thruster,是鲲鹏自研的高性能图检索算法,专为大规模嵌入相似性搜索场景设计。 | KBest文档 仅提供文档仓路径,代码仓待开源。 |
| KScaNN召回算法 | Kunpeng Scalable Nearest Neighbors,鲲鹏向量检索算法,基于倒排索引,结合鲲鹏架构深度优化索引布局和算法流程,充分挖掘芯片潜力形成的向量检索算法。 | KScaNN文档 仅提供文档仓路径,代码仓待开源。 |
| KRL鲲鹏检索算子库 | Kunpeng Retrieval Library,鲲鹏检索算子库,基于鲲鹏平台优化的用于加速向量检索的算子库,KRL可通过替换算子的形式对faiss原生的HNSW、PQFS、IVFPQ和IVFPQFS等算法进行加速。 | KRL文档 仅提供文档仓路径,代码仓待开源。 |
| KVecTurbo召回向量检索加速组件 | KVecTurbo是鲲鹏自研的向量检索加速组件,可对接openGauss向量数据库使用。KVecTurbo通过将高维向量量化压缩,快速获取query的近邻,同时使用SIMD指令集加速距离计算,用于多维向量最近邻搜索。 | KVecTurbo |
| 算法名称 | 描述 | 仓库路径 |
|---|---|---|
| Pfordelta扩展 | 基于鲲鹏SIMD指令集的召回场景倒排索引PForDelta解压缩加速优化。 | |
| hnswlib扩展 | hnswlib召回算法鲲鹏亲和优化,包含通过向量化技术实现FP16高效支持,应用了预取与指令重排等优化策略。 | |
| Faiss扩展 | Faiss检索引擎鲲鹏亲和优化,包含向量化、维度交织查表累加及向量过滤压缩等关键技术。 | |
| RaBitQ扩展 | RaBitQ算法扩展至ARM64(AArch64)架构,引入FP16精度优化、NEON SIMD向量化、汇编级LUT加速、SOAR溢出向量分配、ML自适应nprobe等多项性能优化。 | |
| DiskANN扩展 | DiskANN算法扩展至ARM64(AArch64)架构,引入NEON SIMD向量化、数据布局优化(邻接表驻留内存)、异步IO流水化、精排队列缩减等多项性能优化。 |
| 算法名称 | 描述 | 仓库路径 |
|---|---|---|
| EmbeddingLookup扩展 | 通过编译选项调优、自旋锁优化、内存对齐优化及 ARM SIMD 向量化改造等关键技术,降低实时推荐系统核心模块查表延迟。 |
| 算法名称 | 描述 | 仓库路径 |
|---|---|---|
| KDNN算子库 | Kunpeng Deep Neural Network Library,鲲鹏深度神经网络算子库,结合鲲鹏处理器微架构特性,通过向量化、汇编、算法优化等手段,提升DNN核心算子性能。 | KDNN文档 仅提供文档仓路径,代码仓待开源。 |
| ANNC编译器 | ANNC(Accelerated Neural Network Compiler)是专注于加速神经网络计算的编译器,聚焦于通过计算图优化,高性能融合算子生成和对接技术以及高效代码生成和优化能力,加速推荐和大模型的推理性能,支持主流开源推理框架接入。 | ANNC编译器仓 |
| 算法名称 | 描述 | 仓库路径 |
|---|---|---|
| TensorFlow扩展 | TensorFlow框架鲲鹏亲和优化,包含TF原生算子优化等特性。 | |
| TensorFlow serving扩展 | TensorFlow serving框架鲲鹏亲和优化,包含线程调度优化等特性。 | |
| TVM扩展 | TVM框架鲲鹏亲和优化,包含softmax算子优化等特性。 | |
| oneDNN扩展 | KDNN算子库通过插件化形式对接开源oneDNN库提供完整能力。 | |
| TensorRT-LLM扩展 | 包含算子优化、访存优化、参数配置等特性。 |
搜推模型推理性能Benchmark:包含DLRM、Wide & Deep等多个模型的排序推理性能Benchmark。
鲲鹏BoostKit搜推广使能套件特性列表。
如果发现问题,请进入讨论与我们联系。
使用本领域源码及其附带软件,即视为您已阅读、理解并同意相关软件许可协议条款与条件的约束。
