帶狀稀疏矩陣乘法及高效GPU實現(xiàn)
計算機應用
頁數(shù): 12 2023-06-15
摘要: 稀疏-稠密矩陣乘法(SpMM)廣泛應用于科學計算和深度學習等領域,提高它的效率具有重要意義。針對具有帶狀特征的一類稀疏矩陣,提出一種新的存儲格式BRCV(Banded Row Column Value)以及基于此格式的SpMM算法和高效圖形處理單元(GPU)實現(xiàn)。由于每個稀疏帶可以包含多個稀疏塊,所提格式可看成塊稀疏矩陣格式的推廣。相較于常用的CSR(Compressed Sp... (共12頁)