Open Prokuma opened 4 months ago
このプロジェクトを長期にわたって続けたいとは自分でも思っていないが、NELLの場合はノードの特徴量も構造的にスパースで、疎行列ー疎行列積として扱うコードが多い(PyGもそうだった)
なので、SpGEMMのカーネル取り入れて、NELLの実行速度を上げる試みはやっていいかも
なお、時間かかりそうだったらボツでよし
概要
このプロジェクトを長期にわたって続けたいとは自分でも思っていないが、NELLの場合はノードの特徴量も構造的にスパースで、疎行列ー疎行列積として扱うコードが多い(PyGもそうだった)
なので、SpGEMMのカーネル取り入れて、NELLの実行速度を上げる試みはやっていいかも
要件
なお、時間かかりそうだったらボツでよし