dmlc/dgl

The ALG2 for cusparsespmm under performed when feature size is small

オープン

#4,090 opened on 2022/06/07

 (2 件のコメント) (0 件のリアクション) (0 人の担当者)Python (2,928 件のフォーク)batch import
help wanted

Repository metrics

Stars
 (12,665 個のスター)
PR merge metrics
 (PR metrics pending)

説明

🐛 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

コントリビューターガイド