EnzymeAD / EnzymeAD/Enzyme-JAX
Conv creates allgather
- 主要言語
- MLIR
- スター
- 133
- フォーク
- 53
- 平均マージ
- 1日 10時間
- マージ済み PR（30日）
- 193
説明
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
コントリビューションガイド
このリポジトリのコントリビューションガイドは索引されていません
調査の方向性
まず、`sum_to_conv` を有効にした場合と無効にした場合について、添付された XLA ダンプと XLA 前の MLIR を比較します。Conv 変換によって allgather がどのように生成されるかを追跡し、`sum_to_conv` を有効にしても意図しない allgather が導入されなくなった時点で、この問題が解決されたものとします。
索引モデルが issue の本文から書いたものです。
評価
- 領域
- compilers, machine-learning
- issue の種類
- バグ
- 難易度
- 4/5
- 見積もり時間
- 3〜5日
- 活発さ
- 停滞
- 明瞭さ
- 説明が足りない
- 初心者へのやさしさ
- 25/100