dmlc/dgl

The ALG2 for cusparsespmm under performed when feature size is small

Open

#4,090 opened on 2022幎6月7日

GitHub で芋る
 (2 comments) (0 reactions) (0 assignees)Python (2,928 forks)batch import
help wanted

Repository metrics

Stars
 (12,665 stars)
PR merge metrics
 (30d に merged PR はありたせん)

説明

🐛 Bug

@yzh119 report the ALG2 seems performed badly for cusparsespmm when the feature size is small.

To Reproduce

Steps to reproduce the behavior:

Expected behavior

Environment

  • DGL Version (e.g., 1.0):
  • Backend Library & Version (e.g., PyTorch 0.4.1, MXNet/Gluon 1.3):
  • OS (e.g., Linux):
  • How you installed DGL (conda, pip, source):
  • Build command you used (if compiling from source):
  • Python version:
  • CUDA/cuDNN version (if applicable):
  • GPU models and configuration (e.g. V100):
  • Any other relevant information:

Additional context

コントリビュヌタヌガむド

調査方針
この問題は、小さい特城サむズにおける cusparsespmm アルゎリズム ALG2 のパフォヌマンスリグレッションに぀いお説明しおいたす。調査するには、DGL ゜ヌスコヌド内のスパヌス行列操䜜に関連するファむルにある CUDA カヌネル実装を調べおください。報告者は @yzh119 で、関連する PR やさらなる議論はないため、最初のステップずしお、単玔なベンチマヌクを䜿甚しお問題を再珟し、ALG2 ず他のアルゎリズムを比范するこずです。
技術スタック
pythonpytorch
領域
backendperformance
Issue 皮別
バグ
難床新しい貢献者にずっおの実装難床です。1 は非垞に小さな倉曎、5 は専門的な䜜業です。
3
掚定時間経隓ある貢献者が調査、実装、テスト、pull request 準備にかけるおおよその時間範囲です。
半日
掻動状況Issue が今どれだけ取り組みやすいかを瀺したす。新しい、掻発、叀い、ブロック䞭、たたはメンテナヌ入力埅ちなどです。
アクティブ
明確さ期埅される倉曎、受け入れ条件、次の手順がどれだけ明確かを瀺したす。
おおむね明確
前提条件
PythonCUDAGPU programming
初心者向け床初回貢献者にどれだけ取り組みやすいかを 1-100 で掚定したスコアです。
40