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