开源项目
DeepSelect发布高性能TopK实现
deepseek-ai/DeepSelect
概述
DeepSelect 提供 TopK 算子实现,即从一组得分中快速选择排名最高的若干项。 DeepSelect面向DeepSeek稀疏注意力和采样中的特定TopK负载,项目报告相对torch.topk取得2—20倍局部算子加速。 DeepSelect是DeepSeek稀疏注意力及采样器所用TopK算子的高性能实现,适用于DeepSeek V3.2、V4与V4.1。项目报告相对原生torch.topk取得2—20倍算子加速。