Vulkan 1.4.359 扩展了协作矩阵功能

Vulkan 1.4.359 于 8 月 7 日发布,此次为期一周的 Vulkan API 规范更新带来了一项新的扩展,用于增强协作矩阵的功能。

早在 2023 年的 Vulkan 1.3.255 版本中,就引入了VK_KHR_cooperative_matrix 扩展,用于为 Vulkan 和 SPIR-V IR 提供初步的协作矩阵支持。VK_KHR_cooperative_matrix 可用于机器学习推理、图形后处理、物理模拟等任务。此后,OpenCL 也添加了类似的协作矩阵支持。

Vulkan 1.4.359 新增了 VK_EXT_cooperative_matrix_maintenance1 扩展,对 Vulkan 的协作矩阵支持进行了一系列细微的改进。VK_EXT_cooperative_matrix_maintenance1 新增的功能不仅支持简单的 GEMM 内核,还支持类型/用途转换、归约、逐元素操作等,从而加速了更多特性。新的扩展文本解释如下:

“此扩展在 VK_KHR_cooperative_matrix 中新增的协作矩阵类型基础上,添加了多项新功能。其目标是在简单的 GEMM 内核之外,添加并加速更多功能,包括支持类型/用途转换、归约、逐元素操作以及将元素索引转换为矩阵坐标。

……

此扩展还提供了一个用于查询协作矩阵属性的新命令,未来其他扩展可以轻松地扩展此命令。新查询还包含所需子组大小的输入,以及(当与支持工作组范围协作矩阵的扩展,例如 VK_NV_cooperative_matrix2 一起使用时)本地工作组中的调用次数,从而允许实现公开仅在特定子组大小或本地工作组大小下最优或兼容的属性。”

此次协作矩阵更新由 NVIDIA、高通、英特尔和 Arm 的工程师共同完成。

想要了解更多信息的用户可以通过Vulkan 1.4.359 的此次提交来查看。

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

(0)

相关推荐