Skip to content

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

52 Commits
 
 
 
 
 
 
 
 

Repository files navigation

BoostSRA鲲鹏搜推广使能套件

项目介绍

概述

鲲鹏BoostKit搜推广使能套件旨在为互联网搜索、推荐、广告业务场景提供基于鲲鹏平台的应用层加速能力,组件涵盖召回场景核心检索算法、排序场景模型推理软件框架优化。

架构图

image.png

社区

鲲鹏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

搜推模型推理性能Benchmark:包含DLRM、Wide & Deep等多个模型的排序推理性能Benchmark。

文档

鲲鹏BoostKit搜推广使能套件特性列表

讨论

如果发现问题,请进入讨论与我们联系。

许可协议

使用本领域源码及其附带软件,即视为您已阅读、理解并同意相关软件许可协议条款与条件的约束。

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

Packages

Contributors