EnzymeAD / EnzymeAD/Enzyme-JAX

DUS creates Allgather operations not removed by XLA

Open
#596 10 comments 0 reactions 0 assignees View on GitHub
Dominant language
MLIR
Stars
131
Forks
53
Avg merge
1d 10h
Merged PRs (30d)
193

Description

In [xla_dump.tar.gz](https://github.com/user-attachments/files/19555934/xla_dump.tar.gz) after optimisation we still have plenty collective and allgather operations, even after all the (successful) effort in this package to remove all the transpose operations.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.