スパーシティーの高い行列へ最適化されたカーネルの作成 - Githubissues

NAIST-Archlab / IMAX_GNN

GNNs on IMAX

0 stars 0 forks source link

スパーシティーの高い行列へ最適化されたカーネルの作成 #3

Closed Prokuma closed 11 months ago

Prokuma commented 1 year ago

現状のカーネルは、ユニットを使い切ってSIMDまで駆使してSpMMの演算を行っているため、スパーシティーの低い行列にはあまり向いていない。そのため、スパーシティーの高い行列に最適化されたカーネルの作成を行う。

具体案としては

複数の行ブロックを一回で計算できるようにする
その分一回で計算できる列データが減るが、スパーシティーの高い行列であればこっちのほうが好都合

Prokuma commented 11 months ago

まだやってないがユニット数削減によりある程度目的達成したのでクローズ