Vulkan 1.3.300 提供来自 NVIDIA 的新合作矩阵扩展

Vulkan 1.3.300 于周五首次亮相，其中包含一些修复程序和一个新扩展。

除了各种次要的文档更新/修复外，Vulkan 1.3.300 还引入了 VK_NV_cooperative_matrix2 扩展。2019 年推出的是 NVIDIA 的 VK_NV_cooperative_matrix 扩展，用于允许在 SPIR-V 中协作矩阵类型。Vulkan 1.3.300 是基于 NVIDIA 过去五年的调查结果更新的协作矩阵实现。

Vulkan 协作矩阵支持一直是 NVIDIA 通过 Vulkan API 实现机器学习加速的努力的一部分。更新后的 NVIDIA 合作矩阵扩展描述为：

此扩展在 VK_KHR_cooperative_matrix 中添加的协作矩阵类型的基础上添加了几个新功能。目标是添加和加速简单的 GEMM 内核之外的功能，包括添加对类型/使用转换、缩减、每元素操作和张量寻址的支持，并通过添加对更灵活的矩阵大小的支持，以及通过共享内存使用编译器管理的暂存的工作组范围矩阵来提高可用性和开箱即用的性能。

新功能由 SPV_NV_tensor_addressing 和 SPV_NV_cooperative_matrix2 SPIR-V 扩展定义，可与 GLSL_NV_cooperative_matrix2 GLSL 扩展一起使用。

此扩展包括对枚举实现支持的矩阵类型和维度以及支持哪些特定功能的支持。

对 VK_NV_cooperative_matrix2 的支持已经可以在周五的 Windows 和 Linux 版 NVIDIA Vulkan 驱动程序测试版更新中找到。

通过此 Git 提交了解有关 Vulkan 1.3.300 规范更新的更多详细信息。

转自 Vulkan 1.3.300 Delivers New Cooperative Matrix Extension From NVIDIA – Phoronix