微軟擴大DirectX機器學習布局,讓ML工作負載更直接整合進繪圖管線
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則預計於今年夏季開放私人預覽。
Comments (0)