认识Flash-KMeans:一个IO感知的精确K-Means,在GPU上比FAISS快200倍以上
Flash-KMeans是一个开源的、IO感知的精确K-Means实现,使用Triton GPU内核优化数据流,在不改变数学或近似的情况下实现显著加速。在NVIDIA H200上,它比最佳基线快17.9倍,比cuML快33倍,比FAISS快200倍以上。
认识Flash-KMeans:一个IO感知的精确K-Means,在GPU上比FAISS快200倍以上
本文仅提供摘要
根据来源内容权限,本站仅提供编辑摘要、要点和原文链接。