北京
分享至
IT时代网8月18日消息，一篇聚焦Rust语言GPU编程的研究论文显示，借助LLVM的卸载基础设施，用Rust写出的GPU内核性能已能媲美手写优化的CUDA和HIP C++方案。整套方案完全构建在LLVM后端与Rust编译器内部，不依赖第三方组件。
Rust的所有权系统在编译期就能保证GPU内存安全，借用检查器会提前捕获通信类错误，而不必像CUDA那样留到运行时调试。基准测试在英伟达H100和AMD MI250X两款GPU上完成，采用RAJAPerf套件。在H100上，Rust内核比原生CUDA快11%到慢46%不等，差异主要来自寄存器压力与循环展开的适配度——Rust平均寄存器压力略高（33对28），在对循环展开敏感的微基准里差距最明显。
研究团队表示，后续会通过代码生成优化和中间表示差异适配继续缩小剩余差距。这项研究的看点在于证明内存安全与高性能可以兼得：传统GPU编程为追极致性能常要动用裸指针，而Rust在保留编译期内存安全的同时，跑出了可与手工优化C++竞争的性能。论文由五位学者共同完成，已发表于arXiv平台。
注：本文中包含AI辅助创作的内容。
特别声明：以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布，本平台仅提供信息存储服务。
Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.
