208倍の高速化+5マイクロ秒の予測、H200でKMeansを直接実行して61%のピークFLOPSに到達、バークレーのこの連中はTritonを巧みに操りこなした

原文表示
CoinNetwork
比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による見解の支持や、金融・専門的な助言とみなされるべきものではありません。詳細については免責事項をご覧ください。
  • 報酬
  • コメント
  • リポスト
  • 共有
コメント
コメントを追加
コメントを追加
コメントなし