EnzymeAD / EnzymeAD/Enzyme-JAX
Conv creates allgather
- Lingua principale
- MLIR
- Stelle
- 131
- Fork
- 53
- Merge medio
- 1g 10h
- PR unite (30g)
- 193
Descrizione
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
Guida per i contributori
Nessuna guida per i contributori indicizzata per questo repository
Direzione di ricerca
Inizia confrontando i dump XLA allegati e il MLIR pre-XLA per i casi con `sum_to_conv` abilitato e disabilitato. Traccia come la trasformazione Conv produce l’allgather e considera il problema risolto quando l’abilitazione di `sum_to_conv` non introduce più gli allgather indesiderati.
Scritto dal modello di indicizzazione a partire dal testo della issue.
Valutazione
- Ambito
- compilers, machine-learning
- Tipo di issue
- Bug
- Difficoltà
- 4/5
- Tempo stimato
- 3-5 giorni
- Stato di attività
- Ferma
- Chiarezza
- Da chiarire
- Idoneità per principianti
- 25/100