NAIST-Archlab / IMAX_GNN

GNNs on IMAX
0 stars 0 forks source link

SpGEMMを取り入れる #11

Open Prokuma opened 4 months ago

Prokuma commented 4 months ago

概要

このプロジェクトを長期にわたって続けたいとは自分でも思っていないが、NELLの場合はノードの特徴量も構造的にスパースで、疎行列ー疎行列積として扱うコードが多い(PyGもそうだった)

なので、SpGEMMのカーネル取り入れて、NELLの実行速度を上げる試みはやっていいかも

要件

なお、時間かかりそうだったらボツでよし