208倍提速+5微秒预测,H200上跑KMeans直接干到61%峰值FLOPS,伯克利这帮人把Triton玩出花了

币 界 网
比cuML最高提速208倍,加州大学伯克利分校等开源经典ML加速库FlashLib
币界网称,OneMillion_AI披露的FlashLib开源机器学习加速库由伯克利等团队开发,覆盖15个高层算子,基于Triton与Cutedsl,在H200 GPU对KMeans、KNN等算子实现显著提速,最高208x。与cuML 25.10相比,KMeans26x、KNN19x、HDBSCAN40x、TruncatedSVD208x;KMeans峰值FLOPS达61%,KNN带宽85.2%。此外,FlashLib还提供5微秒内的性能预测API,估算运行时长与显存开销,代码已开源于GitHub。
此页面可能包含第三方内容,仅供参考(非陈述/保证),不应被视为 Gate 认可其观点表述,也不得被视为财务或专业建议。详见声明
  • 赞赏
  • 评论
  • 转发
  • 分享
评论
请输入评论内容
请输入评论内容
暂无评论