EnzymeAD / EnzymeAD/Enzyme-JAX
Conv creates allgather
- Lenguaje dominante
- MLIR
- Estrellas
- 133
- Forks
- 53
- Merge medio
- 1 d 10 h
- PR fusionados (30 d)
- 193
Descripción
On 589fe5bb5967715c5dfa5b49c81151fea25cb75c and with https://github.com/EnzymeAD/Reactant.jl/pull/1138, allgathers are back: [xla_dump.tar.gz](https://github.com/user-attachments/files/19619501/xla_dump.tar.gz) and pre XLA MLIR: https://gist.github.com/giordano/d7e5dd06c664b6ad58ffc5dd9a885482
***Edit***: disabling `sum_to_conv` solves the issue, xla dump without it [xla_dump.tar.gz](https://github.com/user-attachments/files/19619554/xla_dump.tar.gz). Pre XLA MLIR in this case: https://gist.github.com/giordano/8beb071f095f7825199e4a7f447aaad8
Guía de contribución
No hay ninguna guía de contribución indexada para este repositorio
Línea de trabajo
Comienza comparando los volcados de XLA adjuntos y el MLIR previo a XLA para los casos con `sum_to_conv` habilitado y deshabilitado. Rastrea cómo la transformación Conv produce el allgather y considera resuelto el problema cuando habilitar `sum_to_conv` ya no introduzca los allgathers no deseados.
Escrito por el modelo de indexación a partir del texto del issue.
Evaluación
- Área
- compilers, machine-learning
- Tipo de issue
- Error
- Dificultad
- 4/5
- Tiempo estimado
- 3-5 días
- Estado de actividad
- Estancado
- Claridad
- Necesita aclaración
- Aptitud para principiantes
- 25/100