比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 韩国国税厅提议修订《刑事诉讼法》强化

韩国国税厅官员提议修订《刑事诉讼法》，以强化对个人自持虚拟资产的扣押执行规则。今年...

2 韩国股市与纳斯达克100指数相关系数升至

消息，据彭博社披露，韩国综合股价指数与纳斯达克100指数的60日相关系数升至0.46，接近过去...

3 Polymarket：美联储7月29日维持利率不变概率

消息，Polymarket交易者目前预计美联储在7月29日的决策中维持利率不变的概率为93.5%。...

4 长鑫存储巨鲸挂出832万美元卖单，挂单规

消息，长鑫存储巨鲸在HyperLiquid上挂出832万美元的卖单，挂单规模已超过CXMT单日成交额。当前...

5 BTC ETF两周吸引2.73亿美元资金