dmlc/dgl

The ALG2 for cusparsespmm under performed when feature size is small

Ouverte

#4 090 ouverte le 7 juin 2022

Voir sur GitHub
 (2 commentaires) (0 réaction) (0 personne assignée)Python (2 928 forks)batch import
help wanted

Métriques du dépôt

Stars
 (12 665 étoiles)
Métriques de merge PR
 (Métriques PR en attente)

Description

🐛 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

Guide contributeur

Direction de recherche
Le problème décrit une régression de performance dans l'algorithme cusparsespmm ALG2 pour les petites tailles de caractéristiques. Pour enquêter, examinez l'implémentation du noyau CUDA dans le code source de DGL, probablement dans les fichiers liés aux opérations sur matrices creuses. Le rapporteur est @yzh119, et il n'y a pas de PR liés ni de discussion supplémentaire, donc la première étape serait de reproduire le problème en utilisant un benchmark simple et de comparer ALG2 avec d'autres algorithmes.
Stack technique
pythonpytorch
Domaine
backendperformance
Type d'issue
Bug
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
3
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
Une demi journée
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Plutôt claire
Prérequis
PythonCUDAGPU programming
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
40