Skip to content

feat(vector-retrieval): add PurplePulse CUDA retrieval engine - #72

Open
PurplePulse wants to merge 1 commit into
InfiniTensor:2026-summer-projectfrom
PurplePulse:purplepulse-vector-retrieval
Open

PurplePulse wants to merge 1 commit into
InfiniTensor:2026-summer-projectfrom
PurplePulse:purplepulse-vector-retrieval

Conversation

@PurplePulse

Copy link
Copy Markdown

项目信息

  • 选题:09_vector_retrieval
  • 学员 ID:PurplePulse
  • 提交目录:/09_vector_retrieval/PurplePulse/

项目概述

WarpFusion 是面向百万级高维向量检索的单 GPU CUDA 实现,提供 Exact Search 与 IVF-Flat 两条检索路径,支持 FP32/FP16、L2 距离、内积和余弦相似度,以及 K=1/10/50/100。

主要内容

  • 融合距离计算与局部 Top-K,减少 Exact 路径的中间结果写回;
  • 针对 IVF-Flat 实现按桶任务调度和分区局部 Top-K;
  • 提供索引构建、持久化、CPU 参考计算和结果验证;
  • 在 Synthetic-1M、MNIST、SIFT1M 和 Yahoo MiniLM 上完成实验;
  • 提供 FAISS 对照、原始 CSV、中文图表、技术报告和一键复现脚本。

验证情况

  • 全新 CUDA 构建环境下 CTest 18/18 通过;
  • Compute Sanitizer memcheck 记录为 0 errors;
  • Python 工具测试 32 项通过;
  • 发布材料校验通过。

完整的实验口径、性能结果和适用范围见提交目录中的技术报告与复现文档。

- implement GPU Exact and IVF-Flat vector search
- add correctness validation and benchmark tooling
- include benchmark results and technical report
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant