比cuML最高提速208倍,加州大学伯克利分校等开源经典ML加速库FlashLib
币界网消息,OneMillion_AI发文称,加州大学伯克利分校等发布的开源机器学习加速库FlashLib,覆盖15个高层算子,旨在为机器学习工作流与智能体场景提供高性能加速。基于Triton与Cutedsl,FlashLib在H200 GPU上针对KMeans、KNN等算子,较英伟达cuML 25.10实现最高208倍的峰值提速。在H200 GPU测试中,Flash-KMeans效率达到峰值FLOPS的61%,Flash-KNN则达到HBM显存带宽的85.2%。FlashLib在KMeans聚类中提速26倍,在KNN检索中提速19倍,在HDBSCAN聚类中提速40倍,并在TruncatedSVD分解中实现208倍的峰值加速。FlashLib还引入性能预测API,仅用约5微秒的CPU耗时即可精准估算工作流的运行时长与显存开销,已在GitHub开源。(区块链网) [原文链接]
