Vulkan 1.3.300 于上周五发布,其中包括一些修复和一个新扩展。
除了各种小的文档更新/修复之外,Vulkan 1.3.300 还引入了 VK_NV_cooperative_matrix2 扩展。早在 2019 年,英伟达就推出了 VK_NV_cooperative_matrix 扩展,允许在 SPIR-V 中使用合作矩阵类型。随着 Vulkan 1.3.300 的推出,基于英伟达过去五年的研究成果,合作矩阵实现得到了更新。
Vulkan 合作矩阵支持是英伟达通过 Vulkan API 实现机器学习加速的努力的一部分。更新后的英伟达合作矩阵扩展描述如下:
“该扩展以 VK_KHR_cooperative_matrix 中添加的合作矩阵类型为基础,增加了若干新功能。其目的是在简单的 GEMM 内核之外,增加并加速其他功能,包括增加对类型/用途转换、还原、每元素操作和张量寻址的支持,以及通过增加对更灵活的矩阵大小和工作组范围矩阵的支持来提高可用性和开箱即用的性能。
新功能由 SPV_NV_tensor_addressing 和 SPV_NV_cooperative_matrix2 SPIR-V 扩展定义,可与 GLSL_NV_cooperative_matrix2 GLSL 扩展一起使用。
该扩展支持枚举实现所支持的矩阵类型和维数,以及所支持的特定功能。“
上周五发布的 Windows 和 Linux 版英伟达 Vulkan 驱动程序测试版更新已经支持 VK_NV_cooperative_matrix2。
有关 Vulkan 1.3.300 规范更新的更多详细信息,请通过此 Git 提交获取。
本文来自作者投稿,版权归原作者所有。如需转载,请注明出处:https://www.nxrte.com/zixun/53360.html