统一抽象
-
Axe异构布局编译器：跨GPU/TPU/Trainium的统一编程模型，开启机器学习编译新纪元
Axe Layout 的提出，是机器学习系统领域向统一抽象迈进的重要一步。这种统一抽象的威力，在于让开发者能够以接近手工调优代码的性能，轻松编写出高效利用最新 GPU 特性、实现通信计算重叠、并能跨 GPU 和 AI 加速器移植的复杂内核。Axe 不仅仅是一个编译器或 DSL，它更是一种思维范式。它试图弥合高层分布式编程与底层硬件微架构之间的语义鸿沟，为下一…
