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