首页
/ RAFT项目v25.02.00版本发布:分布式计算框架的重大更新

RAFT项目v25.02.00版本发布:分布式计算框架的重大更新

2025-07-09 21:13:01作者:滕妙奇

RAFT是一个开源的GPU加速分布式计算框架,专注于为机器学习和数据分析提供高性能的基础算法库。作为RAPIDS生态系统的重要组成部分,RAFT为GPU加速的机器学习算法提供了底层支持,包括线性代数、最近邻搜索、聚类分析等核心功能。

本次发布的v25.02.00版本带来了多项重要更新和优化,包括性能改进、新功能添加以及一些破坏性变更。下面我们将详细分析这次更新的技术要点。

破坏性变更解析

本次更新包含了三个重要的破坏性变更,开发者需要特别注意:

  1. 开发容器更新:将pip开发容器升级到了UCX 1.18版本。UCX(Unified Communication X)是一个高性能网络通信框架,这次升级可能会影响分布式通信性能和行为。

  2. 日志系统重构:项目全面转向使用rapids-logger作为日志系统。这一变更统一了RAPIDS生态系统的日志接口,但需要开发者调整现有的日志相关代码。

  3. RMM日志适配:为了配合RMM(RAPIDS Memory Manager)的日志系统变更,RAFT也相应调整了日志处理逻辑。这一变更是为了保持与RAPIDS生态其他组件的一致性。

核心功能增强

本次更新在多个核心功能上有所增强:

  1. RMAT矩形生成器优化:修复了RMAT矩形生成器的位顺序问题,确保其输出符合预期。RMAT是一种生成幂律图的随机图模型,广泛用于图算法测试。

  2. Lanczos求解器改进:修复了整数溢出问题。Lanczos算法是求解大型稀疏矩阵特征值问题的重要方法,这一修复提升了算法的数值稳定性。

  3. 位集数据类型支持:新增了对不同数据类型的位集支持,增强了算法的灵活性。位集是表示集合的高效数据结构,广泛应用于图算法和集合运算。

  4. 位集到CSR格式转换:新增了bitset_to_csr功能,实现了位集到压缩稀疏行(CSR)格式的转换。CSR是稀疏矩阵存储的常用格式,这一功能为稀疏矩阵处理提供了更多便利。

性能优化

本次发布包含了多项性能优化措施:

  1. 位图到CSR转换优化:专门针对bitmap_to_csr操作进行了性能优化,提升了转换效率。这对于处理大规模稀疏数据尤为重要。

  2. CUDA 12.8支持:新增了对CUDA 12.8的支持,使项目能够利用最新CUDA版本的性能改进和新特性。

  3. 资源管理改进:引入了raft::device_resources_snmg类型,改进了多GPU环境下的资源管理。这一改进特别针对大规模分布式计算场景。

构建与部署改进

在构建系统和部署方面也有多项改进:

  1. CUDA版本兼容性:移除了cuda-python的上限限制,现在支持12.6.2和11.8.5版本,提高了环境配置的灵活性。

  2. 动态CUDA wheel:在CUDA 11环境下使用动态CUDA wheel,减少了二进制包的大小。

  3. 依赖管理简化:清理了未使用的依赖项(joblib和numba),简化了项目的依赖关系,使部署更加轻量。

  4. libraft wheel支持:引入了libraft的wheel包,简化了Python环境的安装过程。

文档与测试改进

  1. 文档修复:修复了文档构建问题和示例代码错误,提升了文档质量。

  2. 测试结构调整:将测试目录从"test"重命名为"tests",遵循更常见的Python项目结构规范。

  3. CI流程优化:改进了持续集成流程,包括添加了对夜间构建成功率的检查,以及要求对草稿PR运行CI的审批流程。

总结

RAFT v25.02.00版本是一个重要的里程碑更新,在性能、功能和稳定性方面都有显著提升。特别是对最新CUDA版本的支持、日志系统的统一以及多项算法优化,使得RAFT在分布式GPU计算领域继续保持领先地位。开发者升级时需要注意破坏性变更的影响,特别是日志系统的调整和资源管理接口的变化。

这次更新也体现了RAFT项目对代码质量和开发体验的持续关注,通过简化依赖、优化构建流程和改进文档,降低了用户的使用门槛。随着RAPIDS生态系统的不断发展,RAFT作为其基础组件的重要性也将进一步提升。

登录后查看全文
热门项目推荐

项目优选

收起
ohos_react_nativeohos_react_native
React Native鸿蒙化仓库
C++
176
261
RuoYi-Vue3RuoYi-Vue3
🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统
Vue
860
511
ShopXO开源商城ShopXO开源商城
🔥🔥🔥ShopXO企业级免费开源商城系统,可视化DIY拖拽装修、包含PC、H5、多端小程序(微信+支付宝+百度+头条&抖音+QQ+快手)、APP、多仓库、多商户、多门店、IM客服、进销存,遵循MIT开源协议发布、基于ThinkPHP8框架研发
JavaScript
93
15
openGauss-serveropenGauss-server
openGauss kernel ~ openGauss is an open source relational database management system
C++
129
182
openHiTLSopenHiTLS
旨在打造算法先进、性能卓越、高效敏捷、安全可靠的密码套件,通过轻量级、可剪裁的软件技术架构满足各行业不同场景的多样化要求,让密码技术应用更简单,同时探索后量子等先进算法创新实践,构建密码前沿技术底座!
C
259
300
kernelkernel
deepin linux kernel
C
22
5
cherry-studiocherry-studio
🍒 Cherry Studio 是一款支持多个 LLM 提供商的桌面客户端
TypeScript
595
57
CangjieCommunityCangjieCommunity
为仓颉编程语言开发者打造活跃、开放、高质量的社区环境
Markdown
1.07 K
0
HarmonyOS-ExamplesHarmonyOS-Examples
本仓将收集和展示仓颉鸿蒙应用示例代码,欢迎大家投稿,在仓颉鸿蒙社区展现你的妙趣设计!
Cangjie
398
371
Cangjie-ExamplesCangjie-Examples
本仓将收集和展示高质量的仓颉示例代码,欢迎大家投稿,让全世界看到您的妙趣设计,也让更多人通过您的编码理解和喜爱仓颉语言。
Cangjie
332
1.08 K