AMD 在 10 月 30 日发布 ROCm HRX Loom SDK 首个版本 v0.1。这是 ROCm 生态里新构建的编译器与中间件系统,首次实现对 AMD CPU、GPU 及 NPU 的统一编译支持。
它和 HIP 的区别在哪
传统 HIP 路径依赖 LLVM/clang 子进程做编译,启动有开销。Loom 基于原生异步编译架构,可以直接把异步设备内核编译成适用于 AMDGPU 的原生 HSACO ELF64LE 代码对象,单次编译耗时低于 2 毫秒,消除了运行时编译的延迟。
支持什么
当前已支持生成符合 HSA 规范的 AMDGPU 代码供 GPU 执行,同时兼容 SPIR-V 与 Vulkan 标准,保障跨平台部署能力。针对 x86 和 WebAssembly 的支持在推进中,目前已有基础编译功能。指令层面全面支持 MFMA、WMMA 等矩阵计算指令,原生适配 F8、F6、F4 等微量化数值格式。
怎么装
直接 pip install hrx-loom,不需要部署完整 ROCm SDK。这对想快速验证的开发者是很大的便利:过去装 ROCm 要处理驱动、内核模块、用户态运行时一大堆依赖,一条 pip 命令能省掉半天。
自带的两个工具
loom-fetch 用来获取模型资源,loom-serve 通过本地 HTTP 服务快速启用推理。这套组合让你能在一个干净环境里跑通端到端流程。
注意边界
这是 0.1 版本,是 ROCm HRX 开源项目长期演进后的首个正式发布。定位是替代传统 HIP 运行时栈的基础设施,不是给生产环境用的稳定版。真要上生产,先评估你的算子覆盖率和框架兼容性,PyTorch 对 ROCm 的支持成熟度和 CUDA 还有差距。
本文由奇幻维度内容助手整理发布,欢迎在评论区交流你的看法。