Kineto collection of nccl:coalesced don't record metadata

Abierto
#178 0 comentarios 0 reacciones 0 asignados Ver en GitHub

Nadie ha tomado este issue todavía.

Evaluación

Dificultad
4/5
Tiempo estimado
3-5 días
Aptitud para principiantes
35/100
Tipo de issue
Error
Claridad
Necesita aclaración
Estado de actividad
Estancado
Stack tecnológico
python

Línea de trabajo

Comienza reproduciendo la traza de Chakra kineto del script de entrenamiento gpt3/175b_fp8 y compara un all_reduce normal con el nccl coalesced allgather. Sigue la ruta de recopilación de metadatos para ambas operaciones; se considera terminado cuando los kernels coalesced de NCCL incluyen el tamaño de mensaje esperado y los metadatos del grupo de procesos.

Escrito por el modelo de indexación a partir del texto del issue.

Descripción

Chakra's kineto trace add metadata to the NCCL kernels, including message size, PG attributes (name, description, ranks, etc).
However in a trace I just recorded some kernels don't include any metadata.

The trace records gpt3/175b_fp8 training script from NeMo Launcher.

Here is the report of a normal all_reduce operation, we can see info about PG.
image

While here is the report of the nccl coalesced allgather, which displays no metadata:
image

Lenguaje dominante
Python
Estrellas
204
Forks
87
Merge medio
7 d 4 h
PR fusionados (30 d)
3

Preparar el entorno

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Más de mlcommons/chakra

Todos los issues de mlcommons/chakra

Issues similares

Más issues de Python

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.