Vulkan 为 OCP 的微扩展 MX 格式添加扩展,以帮助机器学习

2026年7月3日,Vulkan 1.4.356 发布,其中唯一新增的扩展是 VK_EXT_shader_ocp_microscaling_types,这引起了人们的关注。VK_EXT_shader_ocp_microscaling_types 用于启用开放计算项目 (OCP) 的微缩放 MX 数据类型,从而帮助 Vulkan 处理机器学习工作负载。

OCP 微缩放格式 (MX) 包括 MXFP4、MXFP6、MXFP8 和 MXINT8,旨在提高 AI 训练和推理的效率。VK_EXT_shader_ocp_microscaling_types 增加了对 OCP MX 格式规范中定义的浮点类型的支持,但并未直接支持微缩放张量或对微缩放数据进行操作。

Vulkan 为 OCP 的微扩展 MX 格式添加扩展,以帮助机器学习

出于行业需求,该扩展正在开发中,旨在支持这些新型数据类型,从而提高计算吞吐量并降低内存带宽要求。NVIDIA、三星、Imagination 和 Arm 共同参与了 VK_EXT_shader_ocp_microscaling_types 扩展的开发工作,该扩展已在过去数月内持续推进。

VK_EXT_shader_ocp_microscaling_types 的推出是今日发布的 Vulkan 1.4.356 版本的主要亮点。

本文来自作者投稿,版权归原作者所有。如需转载,请注明出处:https://www.nxrte.com/zixun/69290.html

(0)

相关推荐