最新消息:关注人工智能 AI赋能新媒体运营

微软扩大DirectX机器学习布局,让ML工作负载更直接整合进图形管线

科技资讯 admin 浏览

NVIDIA 公开 DirectX 新一波机器学习相关技术布局,重点包括用于着色器层级运算的 DX Linear Algebra,以及用于完整模型图(Full Model Graphs)执行的 DirectX Compute Graph Compiler。这两项技术的共同方向,是让 Windows 游戏可在 DirectX 绘图工作管线内更直接地执行机器学习工作负载,而不再仅将 DirectX 视为传统绘图 API。

DirectX 这次的调整可分为两个层次,微软说明,第一个层次是将机器学习延伸到着色器内部执行。微软先前已在 Shader Model 6.9 引入 Cooperative Vector,让开发者可在 HLSL 中使用硬件加速的向量乘矩阵运算。现在则进一步推进,新增 DX Linear Algebra,将支持范围从向量-矩阵运算扩大到矩阵-矩阵运算,让降噪、时域超分辨率等更依赖矩阵计算与批量执行的工作负载,也能以同一套 DirectX 程序模型处理。

第二个层次则是将机器学习从着色器层级扩展到完整模型图。微软提出的 DirectX Compute Graph Compiler 是一个全新的 DirectX ML 编译器 API,用途在于让来自现代框架的模型图,先根据设备条件完成图分析、内存规划与运算符融合,再转成可直接整合进 D3D12 队列与命令列表的工作负载。依微软描述,这条路线主要处理较大型、较完整的模型,不再要求开发者将整个 AI 流程拆成个别着色器自行手写。

微软正将 DirectX 中的机器学习能力划分为两条路径:较小、需与传统着色逻辑紧密结合的工作负载,可交由 HLSL 与 DX Linear Algebra 在着色器内直接执行;而较大、具有完整数据流与依赖关系的模型,则交由 DirectX Compute Graph Compiler 负责。微软也表示,后续 DirectX 图形分析与调试工具 PIX 将可在单一捕获流程中,同时查看图形与机器学习工作负载,让开发者能在同一工具链内检查这两类运算。

硬件供应商方面,AMD、Intel、NVIDIA 与 Qualcomm 均表示支持相关技术。微软预告,DX Linear Algebra 将于 4 月进入公开预览,DirectX Compute Graph Compiler 则预计于今年夏季开放私人预览。