EnzymeAD / EnzymeAD/Enzyme-JAX
Conv creates allgather
- Vorherrschende Sprache
- MLIR
- Sterne
- 131
- Forks
- 53
- Ø Merge
- 1 T. 10 Std.
- Gemergte PRs (30 T.)
- 193
Beschreibung
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
Beitragsleitfaden
Für dieses Repository ist kein Beitragsleitfaden indexiert
Rechercherichtung
Beginne damit, die angehängten XLA-Dumps und das Pre-XLA-MLIR für die aktivierten und deaktivierten `sum_to_conv`-Fälle zu vergleichen. Verfolge, wie die Conv-Transformation das allgather erzeugt, und betrachte das Problem als gelöst, wenn das Aktivieren von `sum_to_conv` keine unbeabsichtigten allgathers mehr einführt.
Vom Indexierungsmodell aus dem Issue-Text verfasst.
Bewertung
- Bereich
- compilers, machine-learning
- Issue-Typ
- Bug
- Schwierigkeit
- 4/5
- Geschätzter Aufwand
- 3-5 Tage
- Aktivitätsstatus
- Veraltet
- Klarheit
- Muss geklärt werden
- Anfängerfreundlichkeit
- 25/100