摘要：三對角方程求解器是一種在很多科學(xué)與工程領(lǐng)域廣泛應(yīng)用的數(shù)值計(jì)算核心。目前,CPU、GPU等主流硬件平臺(tái)上都提出了高度優(yōu)化的并行算法,但是對于中國自主研發(fā)的申威26010眾核處理器,還沒有一種算法能有效地利用其獨(dú)特的硬件特性來達(dá)到最大化的性能。提出了一種分布式CR算法swDCR,來求解大量的、規(guī)模不大的三對角方程。該算法對每個(gè)三對角方程使用多個(gè)從核并行求解,通過聯(lián)合多個(gè)從核的緩存使得運(yùn)算過程中所有中間變量都能存儲(chǔ)在緩存中,同時(shí)利用寄存器通信完成核間數(shù)據(jù)的高速傳輸。通過設(shè)計(jì)線程級(jí)數(shù)據(jù)劃分機(jī)制,使得向量化的優(yōu)化效果最大化。swDCR的吞吐率相比主核上的追趕法達(dá)到了單精度43.9倍和雙精度36.7倍的加速,相比從核上的追趕法達(dá)到了單精度和雙精度均2.07倍的加速。該算法在申威26010處理器單個(gè)核組上可以獲得24GB/s的有效帶寬。