比cuML最高提速208倍，加州大学伯克利分校等开源经典ML加速库FlashLib

时间：2026-05-27 20:19:22

　　消息，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开源。

上一篇：0x0ddf...a902：主力空军「pension 返回首页

下一篇：麦迪逊·卡特赖特：美伊谈判前景依然脆弱

热点推荐

1 Robinhood推出AI智能体代购股票服务

消息，美国互联网券商Robinhood宣布推出AI智能体代购股票服务。客户将能够指示AI智能体为他们...

2 Kraken宣布KNTQ现已上线交易

消息，Kraken宣布KNTQ现已上线日起正式开始。用户可通过资金页面添加KNTQ资产，并确保将代币存...

3 麦迪逊·卡特赖特：美伊谈判前景依然脆

消息，澳大利亚联邦银行分析师麦迪逊卡特赖特在一份研报中指出，美伊谈判前景依然脆弱。...

4 比cuML最高提速208倍，加州大学伯克利分校

消息，OneMillion_AI发文称，加州大学伯克利分校等发布的开源机器学习加速库FlashLib，覆盖15个高...

5 0x0ddf...a902：主力空军「pension