dmlc/dgl

The ALG2 for cusparsespmm under performed when feature size is small

Open

#4.090 aberto em 7 de jun. de 2022

Ver no GitHub
 (2 comments) (0 reactions) (0 assignees)Python (2.928 forks)batch import
help wanted

Métricas do repositório

Stars
 (12.665 stars)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 30d)

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

Guia do colaborador

Direção de pesquisa
O problema descreve uma regressão de desempenho no algoritmo cusparsespmm ALG2 para tamanhos de características pequenos. Para investigar, examine a implementação do kernel CUDA no código fonte do DGL, provavelmente em arquivos relacionados a operações de matriz esparsa. O relator é @yzh119, e não há PRs vinculados ou discussões adicionais, então o primeiro passo seria reproduzir o problema usando um benchmark simples e comparar o ALG2 com outros algoritmos.
Pilha de tecnologia
pythonpytorch
Domain
backendperformance
Tipo Issue
Bug
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
3
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
Meio dia
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Principalmente claro
Prerequisites
PythonCUDAGPU programming
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
40