JuliaGPU/GPUArrays.jl

解決的問題

為在 Julia 中建立新 GPU 陣列類型的開發者提供一組標準化的工具與介面。開發者無需為每個新的硬體後端從零開始建構 GPU 陣列功能，而是可以使用此套件加速實作流程，並確保與更廣泛的 Julia 生態系統相容。

工作原理

它作為 Julia 的 AbstractArray 介面在 GPU 上的對應物。定義一組後端開發者必須實作的介面方法（例如 dense_array_type 和 sparse_array_type），以支援密集與稀疏陣列。同時也提供各種稀疏格式（CSC、CSR、COO 和 BSR）的設備端陣列類型，並提供與後端無關的包裝器，支援 GPU 稀疏陣列上的廣播與 mapreduce 功能。

適用對象

此套件主要針對開發 GPU 後端的開發者（例如 CUDA.jl、oneAPI.jl、AMDGPU.jl 或 Metal.jl 的開發者）設計，而非終端使用者。

主要特色

• 與後端無關的包裝器：為 GPU 稀疏陣列提供開箱即用的廣播與 mapreduce 支援。
• 稀疏陣列支援：包含 CSC、CSR、COO 和 BSR 矩陣的設備端類型，以及稀疏向量。
• 介面標準化：定義明確的方法集，用於實作主機端與設備端陣列的轉換。